[CMT][Task]Improve migration wizard UI