modify models.rb