step 4 working (with low accuracy for naive bayes). thank buddha!