Add UML diagram and description