Update Generator.scala