Minor changes in the diagram