diff --git a/docs/_static/flyte.css b/docs/_static/flyte.css
index a0e5225c25..6071b8ab51 100644
--- a/docs/_static/flyte.css
+++ b/docs/_static/flyte.css
@@ -43,6 +43,7 @@ h6 {
   width: 100% !important;
   margin: 0px !important;
   border-radius: 0 !important;
+  border-top: 1px solid var(--color-header-border) !important;
   border-bottom: 1px solid var(--color-header-border) !important;
   background: var(--color-sidebar-background) !important;
   padding: 0 15px !important;
@@ -53,6 +54,16 @@ h6 {
   background: var(--docsearch-searchbox-background) !important;
 }
 
+.DocSearch-Button-Key {
+  border-radius: 3px !important;
+  box-shadow: var(--docsearch-key-shadow) !important;
+  color: var(--docsearch-muted-color) !important;
+  display: flex !important;
+  margin-right: 0.4em !important;
+  padding: 0 0 2px !important;
+  border: 0 !important;
+}
+
 .sidebar-search-container::before {
   content: none;
 }
diff --git a/docs/_templates/base.html b/docs/_templates/base.html
index 867023a4ce..f3c5ef1a49 100644
--- a/docs/_templates/base.html
+++ b/docs/_templates/base.html
@@ -118,8 +118,7 @@
     {# Theme-related JavaScript code #}
     {%- block theme_scripts -%}
     <script src="https://cdn.jsdelivr.net/npm/@docsearch/js@3"></script>
-    <script src="{{ pathto('_static/scripts/algolia-search.js', 1) }}"></script>
-    <script src="{{ pathto('_static/scripts/rate-the-docs.min.js', 1) }}"></script>
+    <script src="{{ pathto('_static/algolia-search.js', 1) }}"></script>
     {%- endblock -%}
     {%- endblock scripts -%}
   </body>
diff --git a/docs/_templates/page.html b/docs/_templates/page.html
index 582062c95e..d078e9dd5e 100644
--- a/docs/_templates/page.html
+++ b/docs/_templates/page.html
@@ -61,6 +61,8 @@
   <div class="main">
     <div class="content">
       <div class="article-container">
+        <!-- add project name to the page so that algolia can extract page category -->
+        <div class="page-category" style="display: none;">{{ project }}</div>
         <a href="#" class="back-to-top muted-link">
           <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24">
             <path d="M13 20h-2V8l-5.5 5.5-1.42-1.42L12 4.16l7.92 7.92-1.42 1.42L13 8v12z"></path>
diff --git a/docs/_templates/search.html b/docs/_templates/sidebar/search.html
similarity index 100%
rename from docs/_templates/search.html
rename to docs/_templates/sidebar/search.html
diff --git a/docs/conf.py b/docs/conf.py
index 60b5bb1232..ed9df55433 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -187,7 +187,6 @@
 # so a file named "default.css" will overwrite the builtin "default.css".
 html_static_path = ["_static"]
 html_css_files = ["custom.css", "flyte.css", "algolia.css"]
-html_js_files = ["algolia-search.js"]
 
 # Custom sidebar templates, must be a dictionary that maps document names
 # to template names.