diff --git a/knn.py b/knn.py
index fece0caf4ff327e411841430f7df1bd0a3cb6b6b..843efe234d45262e4a4ff131500ed796429f3d2b 100644
--- a/knn.py
+++ b/knn.py
@@ -55,12 +55,15 @@ def bench_knn():
         accuracies.append(accuracy)
     
     # Save and show the graph of accuracies
-    fig = plt.figure()
+    plt.figure(figsize=(8, 6))
+    plt.xlabel('K')
+    plt.ylabel('Accuracy')
     plt.plot(k_indices, accuracies)
     plt.title("Accuracy as function of k")
+    plt.legend()
+    plt.grid(True)
     plt.show()
     plt.savefig('image-classification/results/knn_batch_1.png')
-    plt.close(fig)
 
 
 if __name__ == "__main__":
@@ -75,9 +78,4 @@ if __name__ == "__main__":
     # x_test = np.array([[1,2],[4,6]])
     # x_train = np.array([[2,4],[7,2],[4,6]])
     # y_train = [1,2,1]
-    # dist = distance_matrix(x_test,x_train)
-
- 
-
-    
-    
\ No newline at end of file
+    # dist = distance_matrix(x_test,x_train)      
\ No newline at end of file