Update the UML class diagram