Update migration script