Update class diagram design