diff --git a/Jenkinsfile b/Jenkinsfile
index d0dbabb..e059375 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -22,7 +22,11 @@ pipeline {
 		KAGGLE_KEY="$params.KAGGLE_KEY"
 		CUTOFF="$params.CUTOFF"
 	}
-	agent any
+	agent { 
+        dockerfile {
+            additionalBuildArgs '-t ium'
+        }
+    }
 	stages {
 		stage('Stage 1') {
 			steps {
diff --git a/Jenkinsfile2 b/Jenkinsfile2
index f4915f4..aa3dfa9 100644
--- a/Jenkinsfile2
+++ b/Jenkinsfile2
@@ -1,6 +1,6 @@
 pipeline{
 	agent {
-        dockerfile true
+        dockerfile { image 'ium' }
         }
 	parameters {
 		buildSelector(
diff --git a/dataset_download.sh b/dataset_download.sh
index 385f131..d995860 100644
--- a/dataset_download.sh
+++ b/dataset_download.sh
@@ -2,5 +2,5 @@
 kaggle datasets download -d tamber/steam-video-games
 unzip -o steam-video-games.zip
 > data.csv
-head -n $CUTOFF steam-200k.csv >> data.csv
+head -n $CUTOFF steam-200k.csv >> steam-200k.csv