Added the save method