Update the class diagram