修改generator和class