temperary change with current knn code (using demo function)