Update of Diagram Class