tree_height function

Height of a tree.