Update generate_migration.sh