updated uml class diagram with latest changes