Add UML to docs