Update the migration scripts