Added some calculations for Naive Bayes