diff --git a/model.py b/model.py
index 86c8f82..ad48477 100644
--- a/model.py
+++ b/model.py
@@ -21,7 +21,8 @@ preprocessor = ColumnTransformer(
     transformers=[
         ('num', StandardScaler(), ['Age', 'BodyweightKg']),
         ('cat', OneHotEncoder(), ['Sex'])
-    ]
+    ],
+    remainder='passthrough'
 )
 
 pipeline = Pipeline(steps=[
@@ -38,6 +39,6 @@ pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae'])
 X_train_excluded = X_train.iloc[1:]
 y_train_excluded = y_train.iloc[1:]
 
-pipeline.fit(X_train_excluded, y_train_excluded, model__epochs=10, model__validation_split=0.1)
+pipeline.fit(X_train_excluded, y_train_excluded, model__epochs=int(sys.argv[1]), model__validation_split=0.1)
 
 pipeline['model'].save('powerlifting_model.h5')