update CodeGenerator