Made corrections to UML