add belongs_to and has_many