Minor change in class diagram to make implementation feasible