Adding in the UML diagram