Worked some more on the naive bayes.