continue working on tree class