Setting up the training data