Added a bit more functionality to the tree and updated README