Messing around with getting the UML working