working on classify_MNB.py