diff --git a/docker/Dockerfile b/docker/Dockerfile
index f5f6c75..9cb08b0 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -4,3 +4,17 @@ RUN apt update && apt install -y ffmpeg libsm6 libxrender-dev
 RUN pip install Cython
 RUN pip install opencv-python cython_bbox motmetrics numba matplotlib sklearn
 RUN pip install lap
+RUN pip install umap-learn
+ENV NUMBA_CACHE_DIR=/tmp/numba_cache
+RUN pip install bokeh
+RUN pip install ipykernel
+RUN pip install ipython
+# Vscode bug: https://github.com/microsoft/vscode-jupyter/issues/8552
+RUN pip install ipywidgets==7.7.2
+#RUN pip install panel jupyter_bokeh
+
+
+# for bokeh
+EXPOSE 5006
+
+CMD python -m ipykernel_launcher -f $DOCKERNEL_CONNECTION_FILE