update migration.py script