Updated ClassDiagram