Script to generate the dataset