Update generator.rb