Think this should be inside the save() block