Update migration.rb