Update model.rb