Update controller_extension.rb