Added train test for classifier.