diff --git a/Dockerfile b/Dockerfile
index 9e351a1..eada525 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,15 +1,17 @@
 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 pip3 install ipython
+RUN pip3 install torch
+RUN pip3 install numpy
 RUN apt-get install unzip
 
 WORKDIR /app
 
 
-COPY ./travel-insurance-prediction-data.zip ./
-COPY ./preparation.py ./
+COPY ./body-performance-data.zip ./
+COPY ./classification_net.py ./
+