diff --git a/docs/getting_started/simulator.ipynb b/docs/getting_started/simulator.ipynb
index 17f6e29..281eae6 100644
--- a/docs/getting_started/simulator.ipynb
+++ b/docs/getting_started/simulator.ipynb
@@ -2,7 +2,7 @@
   "cells": [
     {
       "cell_type": "code",
-      "execution_count": 13,
+      "execution_count": 56,
       "id": "a8ddc37c-66d2-43e4-8162-6be19a1d70a1",
       "metadata": {},
       "outputs": [],
@@ -13,7 +13,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 14,
+      "execution_count": 57,
       "id": "cab40f99-3460-4497-8b9f-3d669eee1e11",
       "metadata": {},
       "outputs": [],
@@ -86,7 +86,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 15,
+      "execution_count": 58,
       "id": "eb0fadc0-7bb6-4c1d-ae3e-9e16527726ab",
       "metadata": {},
       "outputs": [
@@ -101,13 +101,13 @@
             "State 1 with labels ['carchosen'] and features {}\n",
             "State 2 with labels ['open'] and features {}\n",
             "State 3 with labels ['goatrevealed'] and features {}\n",
-            "State 4 with labels ['target', 'done'] and features {}\n",
+            "State 4 with labels ['done'] and features {}\n",
             "\n",
             "Transitions:\n",
             "0.3333333333333333 -> State 1 with labels ['carchosen'] and features {}\n",
             "1.0 -> State 2 with labels ['open'] and features {}\n",
-            "0.5 -> State 3 with labels ['goatrevealed'] and features {}\n",
-            "1.0 -> State 4 with labels ['target', 'done'] and features {}\n"
+            "1.0 -> State 3 with labels ['goatrevealed'] and features {}\n",
+            "1.0 -> State 4 with labels ['done'] and features {}\n"
           ]
         }
       ],
@@ -130,7 +130,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 16,
+      "execution_count": 59,
       "id": "59ac1e34-866c-42c4-b19b-c2a15c830e2e",
       "metadata": {
         "scrolled": true
@@ -170,14 +170,14 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 17,
+      "execution_count": 60,
       "id": "22871288-755c-463f-9150-f207c2f5c211",
       "metadata": {},
       "outputs": [
         {
           "data": {
             "application/vnd.jupyter.widget-view+json": {
-              "model_id": "dedb21bb36f34f78926a65af349155a0",
+              "model_id": "f71a84e1e87c4ab488ade79e22331df8",
               "version_major": 2,
               "version_minor": 0
             },
@@ -191,7 +191,7 @@
         {
           "data": {
             "application/vnd.jupyter.widget-view+json": {
-              "model_id": "a99224b532bf40dab9b7e7018bed689f",
+              "model_id": "e36300b88941431ca55669e176cda222",
               "version_major": 2,
               "version_minor": 0
             },
@@ -205,7 +205,7 @@
         {
           "data": {
             "application/vnd.jupyter.widget-view+json": {
-              "model_id": "28879179b6684e7789aefe922d77d759",
+              "model_id": "b9586ae0085247b9916bee8f14766c05",
               "version_major": 2,
               "version_minor": 0
             },
@@ -215,26 +215,16 @@
           },
           "metadata": {},
           "output_type": "display_data"
-        },
-        {
-          "data": {
-            "text/plain": [
-              "<stormvogel.visualization.Visualization at 0x118715370>"
-            ]
-          },
-          "execution_count": 17,
-          "metadata": {},
-          "output_type": "execute_result"
         }
       ],
       "source": [
         "#we can also visualize the partial model that we get from the simulator:\n",
-        "show.show(partial_model)\n"
+        "vis = show.show(partial_model)"
       ]
     },
     {
       "cell_type": "code",
-      "execution_count": 18,
+      "execution_count": 61,
       "id": "34d0c293-d090-4e3d-9e80-4351f5fcba62",
       "metadata": {},
       "outputs": [