working on updating UML diagram