Added changes to UML diagram