added full functionality to MigrationFlow