Updated the diagram