Fixing the diagram

Fixed the diagram for the presentation