diff --git a/cleosim/visualization.py b/cleosim/visualization.py
index cc1edc7..da83c11 100644
--- a/cleosim/visualization.py
+++ b/cleosim/visualization.py
@@ -278,3 +278,4 @@ def plot(
         invert_z,
         scatterargs,
     )
+    return fig, ax
diff --git a/pyproject.toml b/pyproject.toml
index 66b3fad..2b2d310 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cleosim"
-version = "0.5.1"
+version = "0.6.0"
 description = "Closed Loop, Electrophysiology, and Optogenetics Simulator: testbed and prototyping kit"
 authors = [
     "Kyle Johnsen <kyle@kjohnsen.org>",