Added UML diagram for the new design