Added tests for search feature