modified trainingset