Updated ClassDiagran