Update controller.rb

Did I get it right?