Added some implementation changes
having an error with building the tree