Cleaned up kNN_benchmark file