updated the diagram