Corrected the diagram