Update controller.rb