diff --git a/Dockerfile b/Dockerfile
index aa1d9e9..4d5e992 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,11 +10,18 @@ WORKDIR /app
 RUN apt-get install -y python3
 RUN apt-get install -y python3-pip
 RUN python3 -m pip install pandas
+RUN python3 -m pip install numpy
+RUN python3 -m pip install torch
 
 COPY ./zadanie1.py ./
 COPY ./Customers.csv ./
+COPY ./train.py ./
+COPY ./test.py ./
 
 RUN chmod +r ./Customers.csv
 RUN chmod +x ./zadanie1.py
+RUN chmod +x ./train.py
+RUN chmod +x ./test.py
 
-CMD python3 ./zadanie1.py
\ No newline at end of file
+CMD python3 ./train.py
+CMD python3 ./test.py
\ No newline at end of file