updates to migration script.