Update to UML

Fully drafted class diagram.