Added search functionality to the tree