Added a class diagram