Minor correction in class diagram