fixed naiveBayes
merged after naiveBayes fix