Edited the UML to reflect discussed changes.