diff --git a/code.py b/code.py
index 04c7cad..44bd89d 100644
--- a/code.py
+++ b/code.py
@@ -25,7 +25,7 @@ def set_colors(labels, colors=colors_options):
 
 
 DataFrame = pd.read_csv('Project2_dataset.csv', header=None)
-DataMatrix = DataFrame.as_matrix()
+DataMatrix = DataFrame.values
 
 normalized_data_matrix = skp.scale(DataMatrix, axis=0)
 
@@ -116,4 +116,4 @@ def set_colors(labels, colors=colors_options):
     ax.scatter(*centroid, c='k', s=150, marker='X')
 plt.title('Subtractive and K-means clusters')
 ax.legend()
-plt.show()
\ No newline at end of file
+plt.show()