Working on the training set