Small UML update