UML : updated class diagram to reflect the recent changes