diff --git a/Dockerfile b/Dockerfile
index 804a083..1d8815f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,3 +4,10 @@ RUN apt-get update && \
     apt-get install -y python3 python3-pip
 
 RUN pip3 install pandas numpy
+
+WORKDIR /app
+
+COPY create-dataset.py /app
+COPY barcelona_weekends.csv /app
+
+CMD ["python3", "create-dataset.py"]
\ No newline at end of file
diff --git a/Jenkinsfile-create-dataset-docker b/Jenkinsfile-create-dataset-docker
index feb5ff4..8f1caa2 100644
--- a/Jenkinsfile-create-dataset-docker
+++ b/Jenkinsfile-create-dataset-docker
@@ -47,7 +47,7 @@ pipeline {
                 }
             }
             steps {
-                sh 'ls'
+                 archiveArtifacts artifacts: 'barcelona_weekends.train.csv, barcelona_weekends.dev.csv, barcelona_weekends.test.csv', fingerprint: true
             }
         }
    }
diff --git a/create-dataset.py b/create-dataset.py
index 281d1be..6329993 100644
--- a/create-dataset.py
+++ b/create-dataset.py
@@ -5,17 +5,17 @@ import numpy as np
 
 cutoff = 10
 
-data = pd.read_csv('./data/barcelona_weekends.csv')
+data = pd.read_csv('./barcelona_weekends.csv')
 data = data.sample(cutoff)
 data = data.iloc[:, 1:]
 
 train_set, dev_set, test_set = np.split(data.sample(frac=1, random_state=42),
                                         [int(.6 * len(data)), int(.8 * len(data))])
 
-train_set.to_csv('data/barcelona_weekends.train.csv', index=False)
-dev_set.to_csv('data/barcelona_weekends.dev.csv', index=False)
-test_set.to_csv('data/barcelona_weekends.test.csv', index=False)
+train_set.to_csv('barcelona_weekends.train.csv', index=False)
+dev_set.to_csv('barcelona_weekends.dev.csv', index=False)
+test_set.to_csv('barcelona_weekends.test.csv', index=False)
 
 
-check = pd.read_csv('./train.csv')
+check = pd.read_csv('./barcelona_weekends.test.csv')
 print(check.head())