Create naivebayes.py