update to dataset generators