made change to Naive bayes probability algorithm: