Adding more features for knn algorithm