Update test cases for Tree class