Messing around with UML output