diff --git a/demo.py b/demo.py
index b5a0f08..d292306 100644
--- a/demo.py
+++ b/demo.py
@@ -4,8 +4,8 @@
 
 #read data
 dataframe = pd.read_fwf('brain_body.txt')
-x_values = dataframe[['Brain']]
-y_values = dataframe[['Body']]
+x_values = dataframe[['Brain']].values.tolist() # Credits : @xyzen
+y_values = dataframe[['Body']].values.tolist()
 
 #train model on data
 body_reg = linear_model.LinearRegression()