Update tree.h

updated tree.h for part2