Update NaiveBayes.py