Update naiveBayes.py