Add tests for searching