Added more changes

Some minor changes to the UML diagram.