Updating Save function