Added the class diagram