added in classifier, and prediction / testing methods