Improved sample, removed bug in naive bayes