added tree_node docs
cleaning up code