A minor tweak to the ClassDiagram