updated migration script