Creating the training dataset