diff --git a/docs/requirements.txt b/docs/requirements.txt
index f18e05b..987ee96 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -1,4 +1,5 @@
 sphinx>=3.0
 sphinx-book-theme
-myst-nb
+myst-parser
 sphinxcontrib-bibtex
+sphinx_togglebutton
diff --git a/docs/source/conf.py b/docs/source/conf.py
index b8a3ffe..809b52d 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -11,7 +11,12 @@
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named "sphinx.ext.*") or your custom
 # ones.
-extensions = ["sphinx_proof", "myst_parser", "sphinxcontrib.bibtex"]
+extensions = [
+    "sphinx_proof",
+    "myst_parser",
+    "sphinxcontrib.bibtex",
+    "sphinx_togglebutton",
+]
 
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ["_templates"]
diff --git a/setup.py b/setup.py
index 0fb25ce..cf96a03 100644
--- a/setup.py
+++ b/setup.py
@@ -33,6 +33,7 @@
         "sphinx-book-theme",
         "sphinxcontrib-bibtex",
         "myst-parser",
+        "sphinx_togglebutton",
     ],
 }