Added comments to the diagram