Class Diagram updated