Dataset generation script ready