diff --git a/docs/theme/main.html b/docs/theme/main.html
index 00d658d..ae4bd3a 100644
--- a/docs/theme/main.html
+++ b/docs/theme/main.html
@@ -1,8 +1,16 @@
 {% extends "base.html" %}
+<!-- version handling -->
+<!-- {% block outdated %}
+You're not viewing the latest version.
+<a href="{{ '../' ~ base_url }}">
+    <strong>Click here to go to latest.</strong>
+</a>
+{% endblock %} -->
+<!-- end version handling -->
 <!-- start announce -->
-{% block announce %}
+<!-- {% block announce %}
 {% include 'announce.html' ignore missing %}
-{% endblock %}
+{% endblock %} -->
 <!-- end announce -->
 {% block content %}
 {{ super() }}
diff --git a/mkdocs.yml b/mkdocs.yml
index d343136..4e5e13f 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -1,4 +1,7 @@
 site_name: Loadax Documentation
+site_url: https://walln.github.io/loadax/
+site_author: Nick Wall
+
 theme:
   name: material
   custom_dir: docs/theme
diff --git a/uv.lock b/uv.lock
index 5a0f5f1..4ffa490 100644
--- a/uv.lock
+++ b/uv.lock
@@ -577,7 +577,7 @@ wheels = [
 
 [[package]]
 name = "loadax"
-version = "0.2.0"
+version = "0.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "flax" },