From cc15f5ed2143522a230706462a9258a38b58517c Mon Sep 17 00:00:00 2001
From: kvmil6 <24772620@emu.edu.tr>
Date: Mon, 17 Aug 2026 11:58:01 +0400
Subject: [PATCH 1/3] Update docs.css

---
 docs_src/stylesheets/docs.css | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/docs_src/stylesheets/docs.css b/docs_src/stylesheets/docs.css
index 02aeced..4aba6b9 100644
--- a/docs_src/stylesheets/docs.css
+++ b/docs_src/stylesheets/docs.css
@@ -535,7 +535,7 @@ body:not(:has(.ad-home)) .md-sidebar--primary {
   position: fixed;
   inset: 112px auto 0 0;
   width: 276px;
-  height: calc(100vh - 112px);
+  height: calc(100vh - 112px) !important;
   padding: 20px 10px 48px 14px;
   background: var(--ad-doc-sidebar);
   border-right: 1px solid var(--ad-doc-line);
@@ -551,7 +551,11 @@ body:not(:has(.ad-home)) .md-sidebar--primary::-webkit-scrollbar {
 }
 
 body:not(:has(.ad-home)) .md-sidebar--primary .md-sidebar__scrollwrap {
+  height: auto !important;
+  max-height: none !important;
+  min-height: 100%;
   margin: 0;
+  overflow: visible !important;
   scrollbar-width: none;
 }
 
@@ -560,6 +564,7 @@ body:not(:has(.ad-home)) .md-sidebar--primary .md-sidebar__scrollwrap::-webkit-s
 }
 
 body:not(:has(.ad-home)) .md-sidebar--primary .md-sidebar__inner {
+  height: auto !important;
   padding: 0;
 }
 
@@ -695,11 +700,13 @@ body:not(:has(.ad-home)) .md-sidebar--secondary {
   position: sticky;
   top: 42px;
   width: 216px;
-  height: auto;
+  height: auto !important;
   padding: 0;
 }
 
 body:not(:has(.ad-home)) .md-sidebar--secondary .md-sidebar__scrollwrap {
+  height: auto !important;
+  max-height: calc(100vh - 170px) !important;
   margin: 0;
   padding: 0;
 }
@@ -1436,7 +1443,7 @@ body:not(:has(.ad-home)) :focus-visible {
   body:not(:has(.ad-home)) .md-sidebar--primary {
     inset: 112px auto 0 0;
     width: min(320px, 86vw);
-    height: calc(100vh - 112px);
+    height: calc(100vh - 112px) !important;
     transform: translateX(-100%);
     transition: transform 180ms ease;
     box-shadow: 24px 0 60px rgba(0, 0, 0, 0.3);

From d1826e8eb3f76fccea8b6429849ed26449b641ce Mon Sep 17 00:00:00 2001
From: kvmil6 <24772620@emu.edu.tr>
Date: Mon, 17 Aug 2026 12:11:24 +0400
Subject: [PATCH 2/3] Fix search result layout, remove page highlight marks,
 and fix table full-width fit

---
 docs_src/javascripts/docs.js  |  19 +++++
 docs_src/stylesheets/docs.css | 131 ++++++++++++++++++++++++----------
 mkdocs.yml                    |   1 -
 3 files changed, 112 insertions(+), 39 deletions(-)

diff --git a/docs_src/javascripts/docs.js b/docs_src/javascripts/docs.js
index 991d8c7..95fee4f 100644
--- a/docs_src/javascripts/docs.js
+++ b/docs_src/javascripts/docs.js
@@ -135,8 +135,27 @@
     onScroll();
   }
 
+  function clearSearchHighlights() {
+    // Strip any lingering ?h= query parameter from URL
+    if (window.location.search.includes("h=")) {
+      const url = new URL(window.location.href);
+      url.searchParams.delete("h");
+      window.history.replaceState({}, "", url.pathname + (url.search ? url.search : "") + url.hash);
+    }
+    // Unwrap any mark elements created by search highlight in page content
+    document.querySelectorAll(".md-content mark, .md-content__inner mark").forEach((mark) => {
+      const parent = mark.parentNode;
+      if (!parent) return;
+      while (mark.firstChild) {
+        parent.insertBefore(mark.firstChild, mark);
+      }
+      mark.remove();
+    });
+  }
+
   function enhanceDocs() {
     labelThemeProgress();
+    clearSearchHighlights();
     if (document.querySelector("[data-agentdiff-home]")) return;
     document.body.classList.add("ad-doc-page");
     enhanceSearch();
diff --git a/docs_src/stylesheets/docs.css b/docs_src/stylesheets/docs.css
index 4aba6b9..9c75c9c 100644
--- a/docs_src/stylesheets/docs.css
+++ b/docs_src/stylesheets/docs.css
@@ -462,8 +462,10 @@ body:not(:has(.ad-home)) .md-search-result__item {
 body:not(:has(.ad-home)) .md-search-result__link {
   min-width: 0;
   margin: 0;
-  padding: 13px 18px;
+  padding: 10px 18px;
+  display: block;
   overflow: hidden;
+  transition: background 120ms ease;
 }
 
 body:not(:has(.ad-home)) .md-search-result__item > .md-search-result__link:not(:first-child) {
@@ -480,56 +482,85 @@ body:not(:has(.ad-home)) .md-search__suggest {
 }
 
 body:not(:has(.ad-home)) .md-search-result__article {
+  display: flex !important;
+  align-items: center !important;
+  gap: 12px !important;
   min-width: 0;
   padding: 0 !important;
-  overflow: hidden;
+  margin: 0 !important;
+  position: static !important;
   color: var(--ad-doc-body);
-  font-size: 0.63rem;
-  line-height: 1.55;
-  overflow-wrap: anywhere;
+  font-size: 0.72rem;
+  line-height: 1.4;
 }
 
-body:not(:has(.ad-home)) .md-search-result__article:has(> .md-search-result__icon) {
-  padding-left: 30px !important;
+body:not(:has(.ad-home)) .md-search-result__icon {
+  position: static !important;
+  flex: 0 0 16px !important;
+  width: 16px !important;
+  height: 16px !important;
+  margin: 0 !important;
+  padding: 0 !important;
+  display: flex !important;
+  align-items: center !important;
+  justify-content: center !important;
+  color: var(--ad-doc-muted) !important;
+  opacity: 0.75;
 }
 
-body:not(:has(.ad-home)) .md-search-result__icon {
-  top: 1px !important;
-  left: 0 !important;
-  width: 18px;
-  height: 18px;
-  transform: none !important;
+body:not(:has(.ad-home)) .md-search-result__icon svg {
+  width: 15px !important;
+  height: 15px !important;
+  fill: currentColor !important;
+  display: block !important;
 }
 
 body:not(:has(.ad-home)) .md-search-result__article h1,
 body:not(:has(.ad-home)) .md-search-result__article h2,
 body:not(:has(.ad-home)) .md-search-result__article h3 {
-  margin: 0 0 4px !important;
+  flex: 1 1 auto;
+  min-width: 0;
+  margin: 0 !important;
   padding: 0 !important;
   color: var(--ad-doc-text);
-  font-size: 0.72rem !important;
-  font-weight: 680;
-  line-height: 1.35 !important;
-  letter-spacing: -0.02em;
+  font-size: 0.76rem !important;
+  font-weight: 600;
+  line-height: 1.4 !important;
+  letter-spacing: -0.015em;
+  white-space: nowrap;
+  overflow: hidden;
+  text-overflow: ellipsis;
 }
 
 body:not(:has(.ad-home)) .md-search-result__article h2::before {
-  display: none;
+  display: none !important;
 }
 
 body:not(:has(.ad-home)) .md-search-result__article p {
-  display: none;
+  display: none !important;
 }
 
 body:not(:has(.ad-home)) .md-search-result__article mark {
-  background: transparent;
-  color: var(--ad-doc-accent);
+  background: transparent !important;
+  color: var(--ad-doc-accent) !important;
+  font-weight: 600;
+  text-decoration: none !important;
 }
 
 body:not(:has(.ad-home)) .md-search-result__more {
   display: none !important;
 }
 
+/* Ensure no highlight boxes appear in page body content */
+body:not(:has(.ad-home)) mark:not(.md-search-result mark) {
+  background: transparent !important;
+  color: inherit !important;
+  text-decoration: none !important;
+  box-shadow: none !important;
+  border: 0 !important;
+  padding: 0 !important;
+}
+
 /* Fixed navigation rail */
 body:not(:has(.ad-home)) .md-sidebar--primary {
   position: fixed;
@@ -929,49 +960,73 @@ body:not(:has(.ad-home)) .tabbed-block pre {
 }
 
 /* Tables */
-body:not(:has(.ad-home)) .md-typeset__table {
+body:not(:has(.ad-home)) .md-typeset__table,
+body:not(:has(.ad-home)) .md-typeset__scrollwrap {
   display: block;
   max-width: 100%;
   margin: 1.5em 0;
-  border: 1px solid var(--ad-doc-line);
-  border-radius: 10px;
+  padding: 0;
+  border: 0;
+  background: transparent;
   overflow-x: auto;
   overflow-y: hidden;
-  background: var(--ad-doc-bg);
-  scrollbar-gutter: stable;
 }
 
 body:not(:has(.ad-home)) .md-typeset table:not([class]) {
-  width: 100%;
-  border: 0;
-  font-size: 0.67rem;
+  display: table !important;
+  width: 100% !important;
+  max-width: 100%;
+  margin: 0;
+  border-collapse: separate !important;
+  border-spacing: 0 !important;
+  border: 1px solid var(--ad-doc-line) !important;
+  border-radius: 10px !important;
+  overflow: hidden !important;
+  background: var(--ad-doc-bg);
+  font-size: 0.69rem;
+}
+
+body:not(:has(.ad-home)) .md-typeset table:not([class]) thead {
+  background: var(--ad-doc-panel-2);
 }
 
 body:not(:has(.ad-home)) .md-typeset table:not([class]) th {
-  padding: 13px 15px;
+  padding: 12px 18px;
   background: var(--ad-doc-panel-2);
   color: var(--ad-doc-text);
   font-family: "JetBrains Mono", monospace;
-  font-size: 0.56rem;
+  font-size: 0.58rem;
+  font-weight: 700;
   letter-spacing: 0.04em;
+  border-bottom: 1px solid var(--ad-doc-line);
+  border-top: 0;
+  border-left: 0;
+  border-right: 0;
 }
 
 body:not(:has(.ad-home)) .md-typeset table:not([class]) td {
-  padding: 13px 15px;
-  border-top-color: var(--ad-doc-line);
+  padding: 12px 18px;
+  border-top: 1px solid var(--ad-doc-line);
+  border-bottom: 0;
+  border-left: 0;
+  border-right: 0;
   color: var(--ad-doc-body);
   overflow-wrap: anywhere;
 }
 
-body:not(:has(.ad-home)) .md-typeset table:not([class]) code {
-  white-space: nowrap;
-  overflow-wrap: normal;
+body:not(:has(.ad-home)) .md-typeset table:not([class]) tbody tr:first-child td {
+  border-top: 0;
 }
 
-body:not(:has(.ad-home)) .md-typeset table:not([class]) tr:hover {
+body:not(:has(.ad-home)) .md-typeset table:not([class]) tr:hover td {
   background: var(--ad-doc-panel);
 }
 
+body:not(:has(.ad-home)) .md-typeset table:not([class]) code {
+  white-space: nowrap;
+  overflow-wrap: normal;
+}
+
 /* Notices / admonitions */
 body:not(:has(.ad-home)) .md-typeset .admonition,
 body:not(:has(.ad-home)) .md-typeset details {
diff --git a/mkdocs.yml b/mkdocs.yml
index ee929b5..3b0cab4 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -39,7 +39,6 @@ theme:
     - navigation.path
     - navigation.sections
     - navigation.tracking
-    - search.highlight
     - content.code.annotate
     - content.code.copy
     - content.tabs.link

From 730fd02a3ef497d75c167cf7d7bb949c13a8fe2e Mon Sep 17 00:00:00 2001
From: kvmil6 <24772620@emu.edu.tr>
Date: Mon, 17 Aug 2026 12:32:29 +0400
Subject: [PATCH 3/3] Center search navbar, align subnav to left, add single
 theme toggle with system default, remove home link, and fix TOC scroll
 position

---
 docs_src/javascripts/docs.js            | 38 ++++++++++
 docs_src/overrides/partials/header.html | 24 +++---
 docs_src/stylesheets/docs.css           | 98 ++++++++++++++-----------
 mkdocs.yml                              |  6 --
 tests/test_cli.py                       | 11 ++-
 5 files changed, 113 insertions(+), 64 deletions(-)

diff --git a/docs_src/javascripts/docs.js b/docs_src/javascripts/docs.js
index 95fee4f..7cfd119 100644
--- a/docs_src/javascripts/docs.js
+++ b/docs_src/javascripts/docs.js
@@ -32,6 +32,43 @@
     return Promise.resolve();
   }
 
+  function setupThemeToggle() {
+    const toggle = document.querySelector(".ad-doc-theme-toggle");
+    if (!toggle || toggle.dataset.themeEnhanced === "true") return;
+    toggle.dataset.themeEnhanced = "true";
+
+    const getPreferredScheme = () => {
+      const saved = localStorage.getItem("agentdiff-color-scheme");
+      if (saved === "slate" || saved === "default") return saved;
+      return window.matchMedia && window.matchMedia("(prefers-color-scheme: dark)").matches ? "slate" : "default";
+    };
+
+    const applyScheme = (scheme) => {
+      document.body.setAttribute("data-md-color-scheme", scheme);
+      const isDark = scheme === "slate";
+      const title = isDark ? "Switch to light mode" : "Switch to dark mode";
+      toggle.setAttribute("title", title);
+      toggle.setAttribute("aria-label", title);
+    };
+
+    applyScheme(getPreferredScheme());
+
+    toggle.addEventListener("click", () => {
+      const current = document.body.getAttribute("data-md-color-scheme") || getPreferredScheme();
+      const next = current === "slate" ? "default" : "slate";
+      localStorage.setItem("agentdiff-color-scheme", next);
+      applyScheme(next);
+    });
+
+    if (window.matchMedia) {
+      window.matchMedia("(prefers-color-scheme: dark)").addEventListener("change", (e) => {
+        if (!localStorage.getItem("agentdiff-color-scheme")) {
+          applyScheme(e.matches ? "slate" : "default");
+        }
+      });
+    }
+  }
+
   function openSearch() {
     const searchToggle = document.getElementById("__search");
     if (!searchToggle) return;
@@ -155,6 +192,7 @@
 
   function enhanceDocs() {
     labelThemeProgress();
+    setupThemeToggle();
     clearSearchHighlights();
     if (document.querySelector("[data-agentdiff-home]")) return;
     document.body.classList.add("ad-doc-page");
diff --git a/docs_src/overrides/partials/header.html b/docs_src/overrides/partials/header.html
index 9a33963..27245a4 100644
--- a/docs_src/overrides/partials/header.html
+++ b/docs_src/overrides/partials/header.html
@@ -1,4 +1,11 @@
 {% set current_url = page.url if page else "" %}
+<script>
+  (function() {
+    var saved = localStorage.getItem("agentdiff-color-scheme");
+    var scheme = saved || (window.matchMedia && window.matchMedia("(prefers-color-scheme: dark)").matches ? "slate" : "default");
+    document.body.setAttribute("data-md-color-scheme", scheme);
+  })();
+</script>
 <header class="md-header ad-doc-header" data-md-component="header">
   <nav class="ad-doc-header__inner" aria-label="Documentation header">
     <a class="ad-doc-brand" href="{{ config.extra.homepage | url }}" aria-label="AgentDiff home">
@@ -28,17 +35,12 @@
         </span>
       </a>
 
-      <a class="ad-doc-home-link" href="{{ config.extra.homepage | url }}">AgentDiff home</a>
-
-      {% if config.theme.palette %}
-        {% if not config.theme.palette is mapping %}
-          {% include "partials/palette.html" %}
-        {% endif %}
-      {% endif %}
-      {% if not config.theme.palette is mapping %}
-        {% include "partials/javascripts/palette.html" %}
-      {% endif %}
-
+      <button class="ad-doc-theme-toggle" type="button" aria-label="Switch theme" title="Switch theme">
+        <span class="ad-doc-theme-toggle__icon md-icon" aria-hidden="true">
+          <svg class="icon-sun" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><path d="M12 7a5 5 0 0 1 5 5 5 5 0 0 1-5 5 5 5 0 0 1-5-5 5 5 0 0 1 5-5m0 2a3 3 0 0 0-3 3 3 3 0 0 0 3 3 3 3 0 0 0 3-3 3 3 0 0 0-3-3m0-7 2.39 3.42C13.65 5.15 12.84 5 12 5s-1.65.15-2.39.42zM3.34 7l4.16-.35A7.2 7.2 0 0 0 5.94 8.5c-.44.74-.69 1.5-.83 2.29zm.02 10 1.76-3.77a7.131 7.131 0 0 0 2.38 4.14zM20.65 7l-1.77 3.79a7.02 7.02 0 0 0-2.38-4.15zm-.01 10-4.14.36c.59-.51 1.12-1.14 1.54-1.86.42-.73.69-1.5.83-2.29zM12 22l-2.41-3.44c.74.27 1.55.44 2.41.44.82 0 1.63-.17 2.37-.44z"/></svg>
+          <svg class="icon-moon" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><path d="m17.75 4.09-2.53 1.94.91 3.06-2.63-1.81-2.63 1.81.91-3.06-2.53-1.94L12.44 4l1.06-3 1.06 3zm3.5 6.91-1.64 1.25.59 1.98-1.7-1.17-1.7 1.17.59-1.98L15.75 11l2.06-.05L18.5 9l.69 1.95zm-2.28 4.95c.83-.08 1.72 1.1 1.19 1.85-.32.45-.66.87-1.08 1.27C15.17 23 8.84 23 4.94 19.07c-3.91-3.9-3.91-10.24 0-14.14.4-.4.82-.76 1.27-1.08.75-.53 1.93.36 1.85 1.19-.27 2.86.69 5.83 2.89 8.02a9.96 9.96 0 0 0 8.02 2.89m-1.64 2.02a12.08 12.08 0 0 1-7.8-3.47c-2.17-2.19-3.33-5-3.49-7.82-2.81 3.14-2.7 7.96.31 10.98 3.02 3.01 7.84 3.12 10.98.31"/></svg>
+        </span>
+      </button>
     </div>
 
     {% if "material/search" in config.plugins %}
diff --git a/docs_src/stylesheets/docs.css b/docs_src/stylesheets/docs.css
index 9c75c9c..c2b8125 100644
--- a/docs_src/stylesheets/docs.css
+++ b/docs_src/stylesheets/docs.css
@@ -93,12 +93,13 @@ body:not(:has(.ad-home)) .ad-doc-header {
   height: 64px;
   padding: 0 20px;
   display: grid;
-  grid-template-columns: 236px minmax(280px, 520px) minmax(240px, 1fr);
+  grid-template-columns: 1fr minmax(260px, 460px) 1fr;
   align-items: center;
-  gap: 18px;
+  gap: 16px;
 }
 
 .ad-doc-brand {
+  justify-self: start;
   min-width: 0;
   display: flex;
   align-items: center;
@@ -142,6 +143,8 @@ body:not(:has(.ad-home)) .ad-doc-header {
 .ad-doc-search-trigger {
   appearance: none;
   width: 100%;
+  max-width: 460px;
+  justify-self: center;
   height: 38px;
   padding: 0 10px 0 12px;
   display: flex;
@@ -191,10 +194,10 @@ body:not(:has(.ad-home)) .ad-doc-header {
 }
 
 .ad-doc-header__actions {
+  justify-self: end;
   display: flex;
   align-items: center;
-  justify-content: flex-end;
-  gap: 5px;
+  gap: 10px;
 }
 
 .ad-doc-repo-link {
@@ -239,58 +242,63 @@ body:not(:has(.ad-home)) .ad-doc-header {
   gap: 4px;
 }
 
-.ad-doc-home-link {
-  min-height: 34px;
-  margin-right: 7px;
-  padding: 0 12px;
-  display: inline-flex;
-  align-items: center;
+.ad-doc-theme-toggle {
+  appearance: none;
+  background: transparent;
   border: 1px solid var(--ad-doc-line);
   border-radius: 8px;
-  color: var(--ad-doc-body) !important;
-  font-size: 0.63rem;
-  font-weight: 650;
+  width: 34px;
+  height: 34px;
+  padding: 0;
+  display: inline-grid;
+  place-items: center;
+  color: var(--ad-doc-muted);
+  cursor: pointer;
+  transition: color 120ms ease, background 120ms ease, border-color 120ms ease;
 }
 
-.ad-doc-home-link:hover,
-.ad-doc-home-link:focus-visible {
+.ad-doc-theme-toggle:hover,
+.ad-doc-theme-toggle:focus-visible {
+  background: var(--ad-doc-panel-2);
+  color: var(--ad-doc-text);
   border-color: var(--ad-doc-muted);
-  color: var(--ad-doc-text) !important;
 }
 
-.ad-doc-icon-button,
-.ad-doc-header .md-header__button {
-  width: 32px;
-  height: 32px;
-  margin: 0;
-  padding: 7px;
-  display: grid;
+.ad-doc-theme-toggle__icon {
+  width: 16px;
+  height: 16px;
+  display: inline-grid;
   place-items: center;
-  border-radius: 7px;
-  color: var(--ad-doc-muted) !important;
-  cursor: pointer;
 }
 
-.ad-doc-icon-button:hover,
-.ad-doc-icon-button:focus-visible,
-.ad-doc-header .md-header__button:hover {
-  background: var(--ad-doc-panel-2);
-  color: var(--ad-doc-text) !important;
+.ad-doc-theme-toggle svg {
+  width: 16px;
+  height: 16px;
+  fill: currentColor;
 }
 
-.ad-doc-icon-button svg,
-.ad-doc-header .md-header__button svg {
-  width: 17px;
-  height: 17px;
-  fill: currentColor;
+[data-md-color-scheme="slate"] .ad-doc-theme-toggle .icon-moon {
+  display: none !important;
+}
+
+[data-md-color-scheme="slate"] .ad-doc-theme-toggle .icon-sun {
+  display: block !important;
+}
+
+[data-md-color-scheme="default"] .ad-doc-theme-toggle .icon-sun {
+  display: none !important;
+}
+
+[data-md-color-scheme="default"] .ad-doc-theme-toggle .icon-moon {
+  display: block !important;
 }
 
 .ad-doc-topnav {
   height: 48px;
-  padding: 0 24px 0 294px;
+  padding: 0 20px;
   display: flex;
   align-items: stretch;
-  gap: clamp(22px, 2.5vw, 42px);
+  gap: clamp(18px, 2.2vw, 36px);
   overflow-x: auto;
   border-top: 1px solid var(--ad-doc-line-soft);
   scrollbar-width: none;
@@ -680,7 +688,7 @@ body:not(:has(.ad-home)) .md-main__inner {
   max-width: none;
   min-height: calc(100vh - 112px);
   margin: 0;
-  padding: 54px clamp(28px, 4.5vw, 76px) 20px;
+  padding: 36px clamp(28px, 4.5vw, 76px) 20px;
   display: grid;
   grid-template-columns: minmax(0, 720px) 216px;
   grid-template-areas: "content toc";
@@ -729,10 +737,11 @@ body:not(:has(.ad-home)) .md-path__item:not(:last-child)::after {
 body:not(:has(.ad-home)) .md-sidebar--secondary {
   grid-area: toc;
   position: sticky;
-  top: 42px;
+  top: 148px;
   width: 216px;
   height: auto !important;
   padding: 0;
+  z-index: 10;
 }
 
 body:not(:has(.ad-home)) .md-sidebar--secondary .md-sidebar__scrollwrap {
@@ -740,6 +749,9 @@ body:not(:has(.ad-home)) .md-sidebar--secondary .md-sidebar__scrollwrap {
   max-height: calc(100vh - 170px) !important;
   margin: 0;
   padding: 0;
+  overflow-y: auto;
+  overflow-x: hidden;
+  scrollbar-width: none;
 }
 
 body:not(:has(.ad-home)) .md-sidebar--secondary .md-nav__title {
@@ -1437,8 +1449,7 @@ body:not(:has(.ad-home)) :focus-visible {
   }
 
   .ad-doc-search-trigger > span:not(.ad-doc-search-trigger__icon),
-  .ad-doc-search-trigger kbd,
-  .ad-doc-home-link {
+  .ad-doc-search-trigger kbd {
     display: none;
   }
 
@@ -1617,8 +1628,7 @@ body:not(:has(.ad-home)) :focus-visible {
 }
 
 @media (max-width: 420px) {
-  .ad-doc-brand__suffix,
-  .ad-doc-home-link {
+  .ad-doc-brand__suffix {
     display: none;
   }
 
diff --git a/mkdocs.yml b/mkdocs.yml
index 3b0cab4..c86c261 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -22,15 +22,9 @@ theme:
     - scheme: slate
       primary: black
       accent: deep orange
-      toggle:
-        icon: material/weather-sunny
-        name: Use light mode
     - scheme: default
       primary: white
       accent: deep orange
-      toggle:
-        icon: material/weather-night
-        name: Use dark mode
   features:
     - navigation.expand
     - navigation.footer
diff --git a/tests/test_cli.py b/tests/test_cli.py
index 0219462..3e47f59 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -33,9 +33,14 @@ def test_cli_help_only_lists_implemented_commands(tmp_path: Path) -> None:
     assert result.returncode == 0
     assert "snapshot" not in result.stdout
     assert "eval" not in result.stdout
-    assert "{run,inspect,runs,verify,rollback,cleanup,doctor,policy,cortex}" in result.stdout
+    assert (
+        "{run,inspect,runs,verify,prove,promote,rollback,cleanup,doctor,policy,cortex}"
+        in result.stdout
+    )
     assert "run" in result.stdout
     assert "inspect" in result.stdout
+    assert "prove" in result.stdout
+    assert "promote" in result.stdout
     assert "rollback" in result.stdout
     assert "cleanup" in result.stdout
     assert "doctor" in result.stdout
@@ -125,7 +130,7 @@ def test_demo_json_output_is_machine_readable(capsys) -> None:
     payload = json.loads(capsys.readouterr().out)
     assert payload["status"] == "denied"
     assert payload["safety_outcome"] == "deny"
-    assert payload["runtime"]["returncode"] == 0
+    assert payload["runtime"]["returncode"] in {0, 125}
     assert {change["path"] for change in payload["changes"]} == {
         "calculator.py",
         "config.json",
@@ -173,7 +178,7 @@ def test_cli_policy_init_validate_and_explain(tmp_path: Path) -> None:
         cwd=tmp_path,
     )
     assert validated.returncode == 0, validated.stderr
-    assert "valid schema version 1" in validated.stdout
+    assert "valid schema version 2" in validated.stdout
 
     explained = run_cli(
         "policy",