diff --git a/docs/_includes/components/phase-banner.html b/docs/_includes/components/phase-banner.html
index fed98d6..5e2662a 100644
--- a/docs/_includes/components/phase-banner.html
+++ b/docs/_includes/components/phase-banner.html
@@ -15,9 +15,11 @@
 
   </div>
 </div>
+<!--
 <script>
   console.log("{{include.auto_hide}}")
   if ("{{include.auto_hide}}" == 'true' && (window.location.hostname.startsWith('notify.nhs.uk') || window.location.hostname.startsWith('127.0.0.2'))) {
     document.getElementById("{{ include.phase_id }}").style.visibility = 'hidden';
   }
 </script>
+-->
diff --git a/docs/_layouts/default.html b/docs/_layouts/default.html
index 78564cd..ab3c23c 100644
--- a/docs/_layouts/default.html
+++ b/docs/_layouts/default.html
@@ -24,8 +24,6 @@
 </head>
 
 <body class="wrap">
-  <script>document.body.className += ' js-enabled' + ('noModule' in HTMLScriptElement.prototype ? ' nhsuk-frontend-supported' : '');</script>
-
   <a class="nhsuk-skip-link" href="#maincontent">Skip to main content</a>
 
   {%- include header.html -%}
@@ -34,10 +32,7 @@
 
   {%- include footer.html -%}
 
-  <script type="module">
-    import { initAll } from '{{site.baseurl}}/assets/js/nhsuk-frontend.min.js'
-    initAll()
-  </script>
+  <script type="module" src="{{site.baseurl}}/assets/js/init.js"></script>
 </body>
 
 </html>
diff --git a/docs/_layouts/page.html b/docs/_layouts/page.html
index 569a764..a5b297c 100644
--- a/docs/_layouts/page.html
+++ b/docs/_layouts/page.html
@@ -121,9 +121,9 @@
           </div>
           <div class="nhsnotify-pane__main-content">
             {% if page.section != undefined %}
-            <heading class="nhsuk-heading-s" style="margin-bottom: 0; color: #4c6272">{{ page.section }}</heading>
+            <span class="nhsuk-caption-xl">{{ page.section }}</span>
             {% endif %}
-            <h1>{{ page.title }}</h1>
+            <h1 class="nhsuk-heading-xl">{{ page.title }}</h1>
             {{ content }}
           </div>
         </div>
diff --git a/docs/assets/css/_nhsnotify-side-nav.scss b/docs/assets/css/_nhsnotify-side-nav.scss
index be60658..427ed6d 100644
--- a/docs/assets/css/_nhsnotify-side-nav.scss
+++ b/docs/assets/css/_nhsnotify-side-nav.scss
@@ -61,6 +61,10 @@
   }
 }
 
+.nhsnotify-side-nav__list-section:not(:first-of-type) {
+  margin-top: 24px;
+}
+
 .nhsnotify-side-nav__item {
   padding: 4px 0 0.5em;
 
diff --git a/docs/assets/js/init.js b/docs/assets/js/init.js
new file mode 100644
index 0000000..601a42b
--- /dev/null
+++ b/docs/assets/js/init.js
@@ -0,0 +1,6 @@
+// Add classes to body for progressive enhancement
+document.body.className += ' js-enabled' + ('noModule' in HTMLScriptElement.prototype ? ' nhsuk-frontend-supported' : '');
+
+// Initialize NHS UK Frontend components
+import { initAll } from './nhsuk-frontend.min.js';
+initAll();
