Update DiagramType.scala