diff --git a/docs/_layouts/default.html b/docs/_layouts/default.html
index cc6b043..113a50a 100644
--- a/docs/_layouts/default.html
+++ b/docs/_layouts/default.html
@@ -3,6 +3,17 @@
 <head>
     <meta charset="UTF-8">
     <title>{{ page.title }}</title>
+    <!-- MathJax configuration -->
+    <script>
+        window.MathJax = {
+            tex: {
+                // Enable both $...$ and \(...\) for inline math,
+                // and both $$...$$ and \[...\] for display math.
+                inlineMath: [['$', '$'], ['\\(', '\\)']],
+                displayMath: [['$$', '$$'], ['\\[', '\\]']]
+            }
+        };
+    </script>
     <!-- Include MathJax -->
     <script src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/3.2.0/es5/tex-mml-chtml.js" async></script>
     <!-- Your CSS and other head elements -->
diff --git a/docs/index.md b/docs/index.md
index a0d1b13..4d4aeb1 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -4,7 +4,6 @@ title: ""Scrambled Attention Guidance"
 toc: true            # Enables automatic table of contents generation (using kramdown)
 ---
 
-[Bacl to main page](..\README.md)
 <!-- Draft Warning Banner -->
 <div style="border: 2px solid red; background-color: #ffe6e6; padding: 15px; text-align: center; margin-bottom: 20px;">
   <h2 style="margin-top: 0;">Draft Document</h2>