Added updated UML diagram.