Add tests for search