Added tests for search functionality