diff --git a/Jenkinsfile b/Jenkinsfile
index 21f4701..0242635 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,5 +1,7 @@
 pipeline {
-    agent any
+    agent {
+        dockerfile true
+    }
     parameters {
         string (
             defaultValue: 'andrzejpreibisz',
diff --git a/main.py b/main.py
index 39f646b..602fd01 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,3 @@
-import csv
-import numpy as np
 import pandas as pd
 from sklearn.model_selection import train_test_split
 import kaggle
diff --git a/main_docker.py b/main_docker.py
index 139f847..b2ad16e 100644
--- a/main_docker.py
+++ b/main_docker.py
@@ -33,3 +33,4 @@ dataset_train["BMI"] = dataset_train["BMI"].apply(lambda x: x/max_bmi)
 dataset_test["BMI"] = dataset_test["BMI"].apply(lambda x: x/max_bmi)
 print(dataset_train["AgeCategory"].value_counts())
 print(dataset_train["BMI"])
+