Add the class diagram