Update naive_bayes_classify.m