Preparing the dataset