change the UML a bit