Add Generator.scala