fixes + start naive bayes test