Migration classes cleanup