Added a function to plot training set and test set pdfs on the same figure