Add vm migration support