Update CodeGenerator.py