adding naive bayes python class