diff --git a/knn.py b/knn.py
index 735e50340a03394d95372e6e6aa466780d1f0e82..31f9a37b7689244950fa4ad532fbd9d2aa14a499 100644
--- a/knn.py
+++ b/knn.py
@@ -1,5 +1,6 @@
 #Libraries
 import numpy as np
+import matplotlib.pyplot as plt
 #Functions
 def distance_matrix(Y , X):
     #This function takes as parameters two matrices X and Y