Add more details to UML diagramm