Naive Bayes Classifier works, has test function