Update Diagram.java