diff --git a/Dockerfile b/Dockerfile
index 3d5fd47..a4758b7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -16,6 +16,7 @@ COPY ./avocado-training.py ./
 COPY ./avocado-evaluation.py ./
 COPY ./sacred-fileobserver.py ./
 COPY ./sacred-mongoobserver.py ./
+COPY ./avocado-mlflow.py ./
 
 RUN chmod +x ./requirements.txt
 RUN chmod +x ./avocado-preprocessing.py 
@@ -23,6 +24,7 @@ RUN chmod +x ./avocado-training.py
 RUN chmod +x ./avocado-evaluation.py
 RUN chmod +x ./sacred-fileobserver.py
 RUN chmod +x ./sacred-mongoobserver.py
+RUN chmod +x ./avocado-mlflow.py
 
 RUN pip3 install -r ./requirements.txt
 # CMD python3 avocado-preprocessing.py
diff --git a/training.Jenkinsfile b/training.Jenkinsfile
index 74ab8c2..94e7502 100644
--- a/training.Jenkinsfile
+++ b/training.Jenkinsfile
@@ -37,6 +37,7 @@ pipeline {
                         sh 'chmod +x avocado-training.py'
                         sh "echo ${EPOCHS} ${BATCH_SIZE}"
                         sh "python3 avocado-training.py ${EPOCHS} ${BATCH_SIZE}"
+                        sh "python3 avocado-mlflow.py ${EPOCHS} ${BATCH_SIZE}"
                  
                     }
                 }
@@ -47,6 +48,8 @@ pipeline {
 		stage('archiveArtifacts') {
 			steps{
 				archiveArtifacts 'avocado_model.h5'
+                archiveArtifacts 'mlruns/**'
+                archiveArtifacts 'avocado_model/**'
 			}
 		}