Updates to migration