small uml changes