diff --git a/runPythonScripts.sh b/runPythonScripts.sh
index 9e1106e..4d88b03 100644
--- a/runPythonScripts.sh
+++ b/runPythonScripts.sh
@@ -1,3 +1,3 @@
 #!/bin/bash
 python3 ./src/task1python.py
-python3 ./src/trainScript.py
\ No newline at end of file
+python3 ./src/trainScript.py 6
\ No newline at end of file
diff --git a/src/trainScript.py b/src/trainScript.py
index 05a2799..b873529 100644
--- a/src/trainScript.py
+++ b/src/trainScript.py
@@ -6,6 +6,9 @@ import numpy as np
 import tensorflow as tf
 from tensorflow.keras import layers
 
+#train params
+numberOfEpoch = sys.argv[1]
+
 cwd = os.path.abspath(os.path.dirname(sys.argv[0]))
 
 pathTrain = cwd + "/../Participants_Data_HPP/Train.csv"
@@ -56,7 +59,7 @@ linear_model.compile(loss = tf.losses.MeanSquaredError(),
                       optimizer = tf.optimizers.Adam(1))
 
 # train model
-history = linear_model.fit(house_price_features, house_price_labels, epochs=5, )
+history = linear_model.fit(house_price_features, house_price_labels, epochs=numberOfEpoch, )
 #callbacks=[cp_callback])
 # print(history)