diff --git a/Dockerfile b/Dockerfile
index 7745198..8498408 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,4 +10,9 @@ RUN mkdir -p /.kaggle && chown -R jenkins /.kaggle
 
 USER jenkins
 
+COPY data_processing.py .
+COPY create_model.py .
+COPY helper.py .
+COPY predict_price.py .
+
 WORKDIR .
\ No newline at end of file
diff --git a/create_model.py b/create_model.py
index 27ca1db..a10823c 100644
--- a/create_model.py
+++ b/create_model.py
@@ -25,5 +25,3 @@ model.compile(optimizer=adam, loss='mean_squared_error')
 model.fit(X_train, Y_train, epochs=20, batch_size=32, validation_data=(X_dev, Y_dev))
 
 model.save('hp_model.h5')
-
-
diff --git a/helper.py b/helper.py
index 0ad19cc..ffe9422 100644
--- a/helper.py
+++ b/helper.py
@@ -1,5 +1,6 @@
 import tensorflow as tf
 
+
 def prepare_tensors(df):
     Y = df["Price"]
     X = df.drop("Price", axis=1)
diff --git a/predict_price.py b/predict_price.py
index 34010a6..83b424a 100644
--- a/predict_price.py
+++ b/predict_price.py
@@ -1,6 +1,5 @@
 import pandas as pd
 from keras.models import load_model
-
 from helper import prepare_tensors
 
 hp_test = pd.read_csv('hp_test.csv')
@@ -11,5 +10,4 @@ model = load_model('hp_model.h5')
 test_predictions = model.predict(X_test)
 
 predictions_df = pd.DataFrame(test_predictions, columns=["Predicted_Price"])
-
 predictions_df.to_csv('hp_test_predictions.csv', index=False)