Update models_controller.rb