Updates to data migration