Add has_one, has_many and belongs_to