added class diagram for prototype