Added the missing UML design