Switched to MultinomialNB