Added the UML file and cleaned up the project