fixed the save feature ... I think