Update to save() method