minor change in NB classifier