class diagram updated