Added a few more tests for search method