Uml diagram
Small changes to uml-diagram