Trying to add the changes for the model and UML