Update DiagramHelper.java