added a class diagram