Updating migrate script