add a class diagram for discussion