Added a couple of tests for the tree height function