Some changes to the migration script