Added tests for search and updated search argument