Update DiagramEditor.java