reorganized to be in same directory as NaiveBayes