code clean up in display_tree
Merge pull request #5 from amWRit/tree

code clean up in display_tree