diff --git a/mkdocs.yml b/mkdocs.yml
index 0673430..0f9a6a1 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -4,5 +4,9 @@ repo_url: https://github.com/det-lab/kaitai_struct_awkward_runtime
 edit_uri: edit/main/docs/
 theme:
   name: material
+  custom_dir: overrides
   features:
-    - content.action.edit
\ No newline at end of file
+    - content.action.edit
+plugins:
+  - mkdocs-jupyter:
+      include_source: true
\ No newline at end of file
diff --git a/overrides/main.html b/overrides/main.html
new file mode 100644
index 0000000..702c96b
--- /dev/null
+++ b/overrides/main.html
@@ -0,0 +1,11 @@
+{% extends "base.html" %}
+
+{% block content %}
+{% if page.nb_url %}
+    <a href="{{ page.nb_url }}" title="Download Notebook" class="md-content__button md-icon">
+        {% include ".icons/material/download.svg" %}
+    </a>
+{% endif %}
+
+{{ super() }}
+{% endblock content %}
diff --git a/pyproject.toml b/pyproject.toml
index aaf304b..68e80f0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -45,7 +45,8 @@ dev = [
   "pytest-cov >=3",
 ]
 docs = [
-  "mkdocs-material"
+  "mkdocs-material",
+  "mkdocs-jupyter"
 ]
 
 [project.urls]