Corrrected error in class diagram