Updated UML with last class