Write test for find method