Fiddling with naive bayes still...