Working on naivebayes.