Update migrate.py