Added the UML we put together