create_tree_layout function

Create a graph layout for plotting