Added the more updated UML