diff --git a/Dockerfile b/Dockerfile
index f51153f..91978ca 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,13 +2,14 @@ FROM python:3.8.0-alpine
 
 RUN apk add --update --no-cache git curl curl-dev vim py3-lxml gcc make libxml2-dev libxslt-dev libc-dev
 
+RUN apk add py2-pip 
+RUN pip2 install cqlsh
+
 RUN addgroup -S appgroup -g 1000 && \
     adduser -u 1000 -S appuser -G appgroup
 
-RUN mkdir /app
+RUN mkdir /app /src
+ADD requirements.txt setup.py ./websucker /src/
 
-ADD requirements.txt /
-RUN CASS_DRIVER_BUILD_CONCURRENCY=4 pip install -r /requirements.txt
-RUN pip install https://git.kemt.fei.tuke.sk/dano/websucker-pip/archive/master.zip
+RUN CASS_DRIVER_BUILD_CONCURRENCY=4 pip install -r /src/requirements.txt
 WORKDIR /app
-ENTRYPOINT ["websuck"]