belongs_to and refactoring