Added updated UML diagram