[Issue #372][Finished] Added the UML