changed the model to belongs_to