Make naive bayes work with 20% accuracy