Move find_references to association module