Updated migration code for clusters and imwg