Update controller_generator.rb