Update save_raw_data to SampleKey api