diff --git a/docs_theme/main.html b/docs_theme/main.html
index e1075bb06..a0f4d4067 100644
--- a/docs_theme/main.html
+++ b/docs_theme/main.html
@@ -198,33 +198,5 @@
     {% for path in config.extra_javascript %}
     <script src="{{ path|url }}" defer></script>
     {% endfor %}
-
-    <script>
-      var shiftWindow = function () {
-        scrollBy(0, -50);
-      };
-
-      if (location.hash) shiftWindow();
-      window.addEventListener("hashchange", shiftWindow);
-
-      $(".dropdown-menu").on("click touchstart", function (event) {
-        event.stopPropagation();
-      });
-
-      // Dynamically force sidenav/dropdown to no higher than browser window
-      $(".side-nav, .dropdown-menu").css(
-        "max-height",
-        window.innerHeight - 130
-      );
-
-      $(function () {
-        $(window).resize(function () {
-          $(".side-nav, .dropdown-menu").css(
-            "max-height",
-            window.innerHeight - 130
-          );
-        });
-      });
-    </script>
   </body>
 </html>
diff --git a/docs_theme/src/js/main.js b/docs_theme/src/js/main.js
index 855bddd65..fa74d1a15 100644
--- a/docs_theme/src/js/main.js
+++ b/docs_theme/src/js/main.js
@@ -13,7 +13,4 @@ import "../../node_modules/bootstrap/js/dist/tooltip";
 
 import "../scss/styles.scss";
 
-const codeBlocks = document.querySelectorAll("pre code");
-codeBlocks.forEach((block) => {
-  block.parentElement.classList.add(["prettyprint", "well"]);
-});
+setupPrettify();
diff --git a/docs_theme/src/scss/styles.scss b/docs_theme/src/scss/styles.scss
index 9501d9c15..098eb9ff2 100644
--- a/docs_theme/src/scss/styles.scss
+++ b/docs_theme/src/scss/styles.scss
@@ -105,7 +105,7 @@ pre code {
 }
 
 .body-content {
-  padding-top: 120px;
+  padding-top: 70px;
   padding-bottom: 70px;
 }
 
@@ -113,6 +113,7 @@ pre code {
   display: flex;
   align-items: center;
   justify-content: end;
+  gap: 8px;
 }
 
 .toclink {
@@ -153,7 +154,7 @@ pre code {
 }
 
 .prettyprint {
-  padding: 8px;
+  padding: 20px;
   background-color: #f7f7f9;
   border: 1px solid #e1e1e8;
 }
diff --git a/mkdocs.yml b/mkdocs.yml
index 4c8bd67c7..b223b0e87 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -14,7 +14,8 @@ watch:
 markdown_extensions:
  - admonition
  - toc:
-    anchorlink: True
+    # anchorlink: True
+    permalink: True
 
 nav:
  - Home: 'index.md'