commented out some lines in naive bayes