Class diagram updated