modify testclass generater.