NB classifier working