trim_tree_at_height function

Trim a rooted tree down to a specific height.