Naive Bayes Classifier in Python (from scratch!)