Update models.rb