use HashSet instead of TreeSet
use correctly lists for the bayes classifier