Add code to save the updated model