split_tree_at_height function

Split a tree into subtrees at a specific height.