updated, changed over to knn classifier