Add test for find method