Added naive bayes to main