changed path to training set