Changes to the class diagram