added new dataset as well as naive bayes fit