diff --git a/Dockerfile b/Dockerfile
index 7474221..f70e7ba 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,13 +3,12 @@ FROM ubuntu:latest
 RUN apt update && apt install -y python3-pip
 #RUN pip3 install kaggle
 RUN pip3 install pandas
+RUN pip3 install sklearn
+RUN pip3 install seaborn
 RUN apt-get install unzip
 
 WORKDIR /app
 
-ENV CUTOFF=100
 
 COPY ./travel-insurance-prediction-data.zip ./
-COPY ./download_data.sh ./
-
-CMD ./download_data.sh
+COPY ./preparation.py ./