update to UML diagram