Updates to migration controller