Added a screenshot of the class diagram