Updated final UML
UML section for writeup