Update CodeGenerator.scala