diff --git a/Project.toml b/Project.toml
index 923cda8..32602a0 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "OptimalTransportNetworks"
 uuid = "e2b46e68-897f-4e4e-ba36-a93c9789fd96"
 authors = ["Sebastian Krantz <sebastian.krantz@graduateinstitute.ch>"]
-version = "0.1.7"
+version = "0.1.6"
 
 [deps]
 Dierckx = "39dd38d3-220a-591b-8e3c-4c3a8c710a94"
diff --git a/docs/make.jl b/docs/make.jl
index 955f862..d73cac5 100644
--- a/docs/make.jl
+++ b/docs/make.jl
@@ -5,7 +5,11 @@ makedocs(
     sitename = "OptimalTransportNetworks.jl",
     modules = [OptimalTransportNetworks],
     checkdocs = :none,
-    format = Documenter.HTML(),
+    format = Documenter.HTML(
+        # Add these options for version switching
+        prettyurls = get(ENV, "CI", nothing) == "true",
+        canonical = "https://SebKrantz.github.io/OptimalTransportNetworks.jl/stable"
+    ),
     pages = [
         "Home" => "index.md",
         "API" => "api.md"
@@ -14,7 +18,7 @@ makedocs(
 
 deploydocs(
     repo = "github.com/SebKrantz/OptimalTransportNetworks.jl.git",
-    branch = "gh-pages",
-    # deploy_config = Dict("DOCUMENTER_KEY" => ENV["DOCUMENTER_KEY"]), 
-    # push_preview = true, ..
+    devbranch = "development",
+    push_preview = true,
+    versions = ["stable" => "v^", "dev" => "development"]
 )
\ No newline at end of file