Added test for search functionality