Refactor find() behavior to use relation module