Added a UML diagram