adding the class diagram