Update Diagram Class