modifications in migration script