Update TaskGenerator.scala