Update migrate script with additional data