clean up NaiveBayes