Update the diagram