Fixed explanation of the loading of the training set