Got the naive Bayes classifier working, I think