Update GraphGen.scala