Update CodeGenVisitor.java