modify make_datasets