diff --git a/spark-qfs-swarm/jupyter-server/Dockerfile b/spark-qfs-swarm/jupyter-server/Dockerfile
index 2a83584..497bafa 100644
--- a/spark-qfs-swarm/jupyter-server/Dockerfile
+++ b/spark-qfs-swarm/jupyter-server/Dockerfile
@@ -2,15 +2,18 @@ FROM qfs-master:latest
 
 #
 # Expected volumes:
-# 	/home/spark/jupyter/notebooks - where the Jupyter notebooks will be persisted
+#   /home/spark/jupyter/notebooks - where the Jupyter notebooks will be persisted
 #   /data/spark - Spark's data directory
 #
 
 USER root
 RUN apt-get install -y g++
-RUN pip3 install notebook==5.7.8 \
-		jupyter_nbextensions_configurator \
-		jupyter_contrib_nbextensions
+RUN pip3 install \
+        notebook==5.7.9 \
+        jupyter_nbextensions_configurator \
+        jupyter_contrib_nbextensions \
+        yapf
+
 COPY start-jupyter.sh /
 
 USER spark
@@ -19,6 +22,7 @@ RUN jupyter nbextensions_configurator enable --user
 RUN jupyter nbextension enable toc2/main
 RUN jupyter nbextension enable codefolding/main
 RUN jupyter nbextension enable execute_time/ExecuteTime
+RUN jupyter nbextension enable code_prettify/code_prettify
 
 RUN mkdir -p /home/spark/jupyter/runtime \
  &&	mkdir -p /home/spark/jupyter/notebooks