diff --git a/Jenkinsfile-dvc b/Jenkinsfile-dvc
index d15e524..0f0d61c 100644
--- a/Jenkinsfile-dvc
+++ b/Jenkinsfile-dvc
@@ -12,10 +12,11 @@ pipeline {
 
         stage("DVC"){
             steps{
-                withCredentials([sshUserPrivateKey(credentialsId: '48ac7004-216e-4260-abba-1fe5db753e18', keyFileVariable: 'IUM_SFTP_KEY')]) {
+                withCredentials(
+                [sshUserPrivateKey(credentialsId: '48ac7004-216e-4260-abba-1fe5db753e18', keyFileVariable: 'IUM_SFTP_KEY', passphraseVariable: '', usernameVariable: '')]) {
+                sh 'dvc remote add -d ium_ssh_remote ssh://ium-sftp@tzietkiewicz.vm.wmi.amu.edu.pl/ium-sftp'
                 sh 'dvc remote modify --local ium_ssh_remote keyfile $IUM_SFTP_KEY'
-                sh "dvc pull"
-                }
+                sh 'dvc pull'}
             }
         }
         stage('Archive Output') {
diff --git a/dvc.yaml b/dvc.yaml
index 5e95f08..8820dc5 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -7,6 +7,8 @@ stages:
       - data.csv
 
   train:
-    cmd: python script5.py
+    cmd: python script5_3.py
     deps:
       - data.csv
+    outs:
+      - model.pth