made small changes to the uml