Modifications of the class diagram