modified class diagram for presentation