Added get_tree_height