Working naive Bayes, but results are questionable