KNN first version!
remove main
Naivebayes