Update CodeGen