diff --git a/docs/docs/examples/example-1-floquet.py b/docs/docs/examples/example-1-floquet.py
index 0b2f073..936dbe0 100644
--- a/docs/docs/examples/example-1-floquet.py
+++ b/docs/docs/examples/example-1-floquet.py
@@ -29,6 +29,9 @@
 import numpy as np
 import matplotlib.pyplot as plt
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
+
 # %% [markdown]
 # ## Define the program.
 # For the floquet protocol we keep We do the same Rabi drive but allow the detuning to
diff --git a/docs/docs/examples/example-1-rabi.py b/docs/docs/examples/example-1-rabi.py
index c3fc928..ebd42e4 100644
--- a/docs/docs/examples/example-1-rabi.py
+++ b/docs/docs/examples/example-1-rabi.py
@@ -29,6 +29,9 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
+
 # %% [markdown]
 # ## Define the program.
 # Below we define program with one atom, with constant detuning but variable Rabi
diff --git a/docs/docs/examples/example-1-ramsey.py b/docs/docs/examples/example-1-ramsey.py
index ee97b98..f66c22f 100644
--- a/docs/docs/examples/example-1-ramsey.py
+++ b/docs/docs/examples/example-1-ramsey.py
@@ -32,6 +32,8 @@
 import numpy as np
 import matplotlib.pyplot as plt
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
 
 # %% [markdown]
 # ## Define the program.
diff --git a/docs/docs/examples/example-2-multi-qubit-blockaded.py b/docs/docs/examples/example-2-multi-qubit-blockaded.py
index 8c8f6f2..16c6366 100644
--- a/docs/docs/examples/example-2-multi-qubit-blockaded.py
+++ b/docs/docs/examples/example-2-multi-qubit-blockaded.py
@@ -31,6 +31,9 @@
 
 import os
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
+
 # %% [markdown]
 # ## Defining the Geometry
 # We will start by defining the geometry of the atoms. The idea here is to cluster
diff --git a/docs/docs/examples/example-2-nonequilibrium-dynamics-blockade-radius.py b/docs/docs/examples/example-2-nonequilibrium-dynamics-blockade-radius.py
index 8b8620d..2137169 100644
--- a/docs/docs/examples/example-2-nonequilibrium-dynamics-blockade-radius.py
+++ b/docs/docs/examples/example-2-nonequilibrium-dynamics-blockade-radius.py
@@ -31,6 +31,9 @@
 
 import os
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
+
 # %% [markdown]
 # ## Program Definition
 # We will start by defining a program. We set up a chain of two atoms
diff --git a/docs/docs/examples/example-2-two-qubit-adiabatic.py b/docs/docs/examples/example-2-two-qubit-adiabatic.py
index 3396573..3824901 100644
--- a/docs/docs/examples/example-2-two-qubit-adiabatic.py
+++ b/docs/docs/examples/example-2-two-qubit-adiabatic.py
@@ -29,6 +29,8 @@
 
 import os
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
 
 # %% [markdown]
 # ## Defining the Program
diff --git a/docs/docs/examples/example-3-2d-ordered-state.py b/docs/docs/examples/example-3-2d-ordered-state.py
index e39f8a4..d8513b0 100644
--- a/docs/docs/examples/example-3-2d-ordered-state.py
+++ b/docs/docs/examples/example-3-2d-ordered-state.py
@@ -34,6 +34,9 @@
 
 import os
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
+
 # This tells Bokeh to display output in the notebook
 # versus opening a browser window
 output_notebook()
diff --git a/docs/docs/examples/example-3-time-sweep.py b/docs/docs/examples/example-3-time-sweep.py
index ce7fcda..bff334b 100644
--- a/docs/docs/examples/example-3-time-sweep.py
+++ b/docs/docs/examples/example-3-time-sweep.py
@@ -31,6 +31,9 @@
 import os
 import matplotlib.pyplot as plt
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
+
 # %% [markdown]
 # ## Program Definition We define a program where our geometry is a chain of 11 atoms
 # with a distance of 6.1 micrometers between atoms.
diff --git a/docs/docs/examples/example-4-quantum-scar-dynamics.py b/docs/docs/examples/example-4-quantum-scar-dynamics.py
index 61252da..d684c66 100644
--- a/docs/docs/examples/example-4-quantum-scar-dynamics.py
+++ b/docs/docs/examples/example-4-quantum-scar-dynamics.py
@@ -49,6 +49,8 @@
 import numpy as np
 import os
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
 
 n_atoms = 11
 atom_spacing = 6.1
diff --git a/docs/docs/examples/example-5-MIS-UDG.py b/docs/docs/examples/example-5-MIS-UDG.py
index 9989c73..677dd45 100644
--- a/docs/docs/examples/example-5-MIS-UDG.py
+++ b/docs/docs/examples/example-5-MIS-UDG.py
@@ -44,6 +44,9 @@
 import os
 import matplotlib.pyplot as plt
 
+if not os.path.isdir("data"):
+    os.mkdir("data")
+
 # setting the seed
 rng = np.random.default_rng(1234)