diff --git a/.gitignore b/.gitignore
index 8ae9691..37d8bae 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,4 +2,6 @@
 test*.ipynb
 .ipynb_checkpoints
 Pipfile.lock
-explorer/*/*.ndjson
\ No newline at end of file
+explorer/*/*.ndjson
+_site
+Gemfile.lock
\ No newline at end of file
diff --git a/Gemfile b/Gemfile
new file mode 100644
index 0000000..68ec4df
--- /dev/null
+++ b/Gemfile
@@ -0,0 +1 @@
+gem "just-the-docs"
\ No newline at end of file
diff --git a/_config.yml b/_config.yml
index 3e40216..320c932 100644
--- a/_config.yml
+++ b/_config.yml
@@ -1,5 +1,6 @@
+theme: just-the-docs
 remote_theme: just-the-docs/just-the-docs
-title: "DiffusionDB"
+title: 'DiffusionDB'
 
 # Enable or disable the site search
 # Supports true (default) or false
@@ -30,7 +31,6 @@ search:
 
 color_scheme: light
 
-
 # External navigation links
 nav_external_links:
   - title: DiffusionDB on GitHub
diff --git a/_includes/head_custom.html b/_includes/head_custom.html
index 4695bb0..642edaf 100644
--- a/_includes/head_custom.html
+++ b/_includes/head_custom.html
@@ -1,3 +1,7 @@
 <link rel="shortcut icon" type="image/png" href="favicon.ico" />
 <meta property="og:image" content="https://i.imgur.com/vNrixuh.png" />
 <meta property="twitter:image" content="https://i.imgur.com/vNrixuh.png" />
+<script
+  type="module"
+  src="https://unpkg.com/recommender-overlay/dist/recommender-overlay.es.js"
+></script>
diff --git a/_layouts/default.html b/_layouts/default.html
new file mode 100644
index 0000000..81f3c55
--- /dev/null
+++ b/_layouts/default.html
@@ -0,0 +1,52 @@
+---
+layout: table_wrappers
+---
+
+<!DOCTYPE html>
+
+<html lang="{{ site.lang | default: 'en-US' }}">
+{% include head.html %}
+<body>
+  <a class="skip-to-main" href="#main-content">Skip to main content</a>
+  {% include icons/icons.html %}
+  {% include components/sidebar.html %}
+  <div class="main" id="top">
+    {% include components/header.html %}
+    <div class="main-content-wrap">
+      {% include components/breadcrumbs.html %}
+      <div id="main-content" class="main-content">
+        <main>
+          {% if site.heading_anchors != false %}
+            {% include vendor/anchor_headings.html html=content beforeHeading="true" anchorBody="<svg viewBox=\"0 0 16 16\" aria-hidden=\"true\"><use xlink:href=\"#svg-link\"></use></svg>" anchorClass="anchor-heading" anchorAttrs="aria-labelledby=\"%html_id%\"" %}
+          {% else %}
+            {{ content }}
+          {% endif %}
+
+          {% if page.has_children == true and page.has_toc != false %}
+            {% include components/children_nav.html %}
+          {% endif %}
+        </main>
+        {% include components/footer.html %}
+      </div>
+    </div>
+    {% if site.search_enabled != false %}
+      {% include components/search_footer.html %}
+    {% endif %}
+  </div>
+
+  {% if site.mermaid %}
+    {% include components/mermaid.html %}
+  {% endif %}
+
+  <recommender-overlay
+    my-brand="DiffusionDB"
+    brands-to-ignore="CNN 101"
+    recommendation-count="10"
+    similar-candidate-count="15"
+    position-left="30"
+    display-delay="15000"
+    homepage-url="https://poloclub.github.io/"
+  ></recommender-overlay>
+
+</body>
+</html>
