diff --git a/Code/Day 1_Data_Preprocessing.py b/Code/Day 1_Data_Preprocessing.py
index d81deb2..9a76411 100644
--- a/Code/Day 1_Data_Preprocessing.py	
+++ b/Code/Day 1_Data_Preprocessing.py	
@@ -19,7 +19,7 @@ print(Y)
 # from sklearn.impute import SimpleImputer
 # imputer = SimpleImputer(missing_values="NaN", strategy="mean")
 from sklearn.preprocessing import Imputer
-imputer = Imputer(missing_values = "NaN", strategy = "mean", axis = 0)
+imputer = Imputer(missing_values = np.nan, strategy = "mean", axis = 0)
 imputer = imputer.fit(X[ : , 1:3])
 X[ : , 1:3] = imputer.transform(X[ : , 1:3])
 print("---------------------")
@@ -33,7 +33,7 @@ from sklearn.preprocessing import LabelEncoder, OneHotEncoder
 labelencoder_X = LabelEncoder()
 X[ : , 0] = labelencoder_X.fit_transform(X[ : , 0])
 #Creating a dummy variable
-onehotencoder = OneHotEncoder(categorical_features = [0])
+onehotencoder = OneHotEncoder('auto')
 X = onehotencoder.fit_transform(X).toarray()
 labelencoder_Y = LabelEncoder()
 Y =  labelencoder_Y.fit_transform(Y)