UML work
Added working UML