changed trainClassifier so it returns prtDS for testing