add plot for the trainingset
merge