Edited the UML diagram of the exercise