Added timeout to dataset generation