Add workloads support to migration cli script