Python nltk and Naive Bayes model