Add test cases for searching in the tree