Updates to migration scripts