Added test for search function.