diff --git a/Dockerfile b/Dockerfile
index 9937241..cf992bb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,8 @@
 FROM ubuntu:latest
 
 RUN apt update && apt install -y python3 \
-    python3-pip
+    python3-pip \
+    git
 
 WORKDIR /code
 
diff --git a/Jenkins/Jenkinsfile.training b/Jenkins/Jenkinsfile.training
index 0e5f52d..7c60cce 100644
--- a/Jenkins/Jenkinsfile.training
+++ b/Jenkins/Jenkinsfile.training
@@ -47,7 +47,7 @@ node {
         stage('Archive artifacts') {
             archiveArtifacts "model/neural_net"
             archiveArtifacts "my_runs/**"
-            archiveArtifacts "/code/mlruns/**"
+            archiveArtifacts "mlruns/**"
         }
         }
     } catch (e) {
diff --git a/Scripts/train_neural_network.py b/Scripts/train_neural_network.py
index 12189e7..1a00141 100644
--- a/Scripts/train_neural_network.py
+++ b/Scripts/train_neural_network.py
@@ -19,7 +19,7 @@ import logging
 logging.basicConfig(level=logging.WARN)
 logger = logging.getLogger(__name__)
 
-# mlflow.set_tracking_uri("http://172.17.0.1:5000")
+mlflow.set_tracking_uri("http://172.17.0.1:5000")
 mlflow.set_experiment("s444452")
 
 ex = Experiment(name='s444452_fake_job_classification_training', save_git_info=False)