diff --git a/docker-compose.yml b/docker-compose.yml
index 001f90c..a1e36f1 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -19,7 +19,7 @@ services:
       - "9077:9077"
       - "10000:10000"
   jupyter: 
-    image: hadesarchitect/jupyter-pyspark-cassandra
+    build: ./pyspark-cassandra
     volumes:
       - ./jupyter:/home/jovyan
     ports:
diff --git a/pyspark-cassandra/Dockerfile b/pyspark-cassandra/Dockerfile
new file mode 100644
index 0000000..8f88b92
--- /dev/null
+++ b/pyspark-cassandra/Dockerfile
@@ -0,0 +1,4 @@
+FROM jupyter/pyspark-notebook
+USER root
+RUN pip install cassandra-driver
+USER jovyan