diff --git a/Jenkinsfile_create_dataset b/Jenkinsfile_create_dataset
index 0b4e701..2d11031 100644
--- a/Jenkinsfile_create_dataset
+++ b/Jenkinsfile_create_dataset
@@ -1,6 +1,11 @@
 pipeline {
     agent any
-  //  properties([parameters([string(defaultValue: '20000', description: 'numbers of data entries to keep in train.csv', name: 'CUTOFF', trim: true)])])
+    parameters {
+        string(defaultValue: '20000',
+            description: 'numbers of data entries to keep in train.csv',
+            name: 'CUTOFF',
+            trim: true)
+    }
     stages {
         stage('sh: Shell Script') {
             steps {
@@ -8,7 +13,7 @@ pipeline {
                 "KAGGLE_KEY=${params.KAGGLE_KEY}" ]) {
                     sh 'chmod +x script-zadanie-2-4.sh'
                     sh './script-zadanie-2-4.sh'
-                   // sh 'head -n ${CUTOFF} train.csv > train.csv'
+                    sh 'head -n ${params.CUTOFF} train.csv > train.csv'
                 }
             }
         }