Add UML Diagram generation to tests.