Update class diagram