diff --git a/function.py b/function.py
index e3f32e3..6e7abe7 100644
--- a/function.py
+++ b/function.py
@@ -1,5 +1,6 @@
 # function.py
 import numpy as np  
+import sklearn
 from sklearn.preprocessing import StandardScaler   
 from numpy.linalg import inv, LinAlgError  
 from scipy.spatial.distance import mahalanobis