Add generator.rb