Update setup_generator.rb