Add tests for find