Update controller.rb

Did I get any close to what you asked? :)