Update migration_generator.rb