diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index aa917c8..b301a65 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -207,7 +207,7 @@
   <component name="PropertiesComponent">
     <property name="WebServerToolWindowFactoryState" value="false" />
     <property name="aspect.path.notification.shown" value="true" />
-    <property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588354194556" />
+    <property name="com.android.tools.idea.instantapp.provision.ProvisionBeforeRunTaskProvider.myTimeStamp" value="1588354296835" />
     <property name="go.gopath.indexing.explicitly.defined" value="true" />
     <property name="nodejs_interpreter_path.stuck_in_default_project" value="undefined stuck path" />
     <property name="nodejs_npm_path_reset_for_default_project" value="true" />
diff --git a/serve.py b/serve.py
index 2ea0d23..1dfd2a9 100644
--- a/serve.py
+++ b/serve.py
@@ -14,7 +14,7 @@ import csv
 
 
 class Config:
-	feature_columns = list(range(0, 1))
+	feature_columns = list(range(0, 2))
 	label_columns = [1]
 	feature_and_label_columns = feature_columns + label_columns
 	label_in_feature_columns = (lambda x, y: [x.index(i) for i in y])(feature_columns, label_columns)
@@ -184,7 +184,7 @@ def train_models():
 	if train_X.shape[0] < 200:
 		config.batch_size = 16
 
-	train(config, train_X, train_Y, valid_X, valid_Y)
+	train(config, train_X[:, :1], train_Y, valid_X[:, :1], valid_Y)
 
 	lock.release()