diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index cae67f7..76a6275 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -39,6 +39,7 @@ pipeline {
 		stage('Archive artifacts') {
 			steps{
 				archiveArtifacts 'wine_model.h5'
+                archiveArtifacts 'my_runs/**'
 			}
 		}
 	}
diff --git a/Zadanie_06_training.py b/Zadanie_06_training.py
index 6b908a8..1a36a82 100644
--- a/Zadanie_06_training.py
+++ b/Zadanie_06_training.py
@@ -67,10 +67,16 @@ def prepare_model(epoch_param, batch_param, _run):
 
     NeuralModel.save('wine_model.h5')
 
+    #TO TYLKO NA POTRZEBY ZADANIA
+    _run.info["Accuracy dla zbioru test (tylko na potrzeby zadania)"] = str(NeuralModel.predict(scaled_x))
+
+
+
+
 @ex.main
 def my_main(epoch_param, batch_param):
     print(prepare_model())
 
 
 r = ex.run()
-ex.add_artifact("saved_model/saved_model.pb")
\ No newline at end of file
+ex.add_artifact("wine_model.h5")
\ No newline at end of file