Updated naive bayes code and results