Updated UML image

Class Diagram