Update before_migrate.rb