Update class diagram with final code