Tree Problem 4 + Unit Test