diff --git a/docs/getting_started/01_study.ipynb b/docs/getting_started/01_study.ipynb
index b76699d..23173e4 100644
--- a/docs/getting_started/01_study.ipynb
+++ b/docs/getting_started/01_study.ipynb
@@ -161,263 +161,34 @@
         "scrolled": true
       },
       "outputs": [
+        {
+          "data": {
+            "application/vnd.jupyter.widget-view+json": {
+              "model_id": "be7c1a38bc504a67abc215743d681813",
+              "version_major": 2,
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Output()"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        },
         {
           "data": {
             "text/html": [
-              "\n",
-              "          <iframe\n",
-              "                id=\"studysFvnWfhilS\"\n",
-              "                width=\"820\"\n",
-              "                height=\"620\"\n",
-              "                frameborder=\"0\"\n",
-              "                srcdoc=\"\n",
-              "&lt;!DOCTYPE html&gt;\n",
-              "&lt;html lang=&quot;en&quot;&gt;\n",
-              "  &lt;head&gt;\n",
-              "    &lt;title&gt;Network&lt;/title&gt;\n",
-              "    &lt;script\n",
-              "      type=&quot;text/javascript&quot;\n",
-              "      src=&quot;https://unpkg.com/vis-network/standalone/umd/vis-network.min.js&quot;\n",
-              "    &gt;&lt;/script&gt;\n",
-              "    &lt;style type=&quot;text/css&quot;&gt;\n",
-              "      #mynetwork {\n",
-              "        \n",
-              "        width: 800px;\n",
-              "        height: 600px;\n",
-              "        border: 1px solid lightgray;\n",
-              "        \n",
-              "      }\n",
-              "    &lt;/style&gt;\n",
-              "  &lt;/head&gt;\n",
-              "  &lt;body&gt;\n",
-              "    &lt;div id=&quot;mynetwork&quot;&gt;&lt;/div&gt;\n",
-              "    &lt;script type=&quot;text/javascript&quot;&gt;\n",
-              "        \n",
-              "        var nodes = new vis.DataSet([{ id: 0, label: `init`, group: &quot;states&quot;, x: -27, y: -203 },\n",
-              "{ id: 1, label: `studied\n",
-              "\u20ac\tR: 0`, group: &quot;states&quot;, x: -84, y: -6 },\n",
-              "{ id: 2, label: `didn&#x27;t study\n",
-              "\u20ac\tR: 15`, group: &quot;states&quot;, x: 57, y: -10 },\n",
-              "{ id: 3, label: `pass test\n",
-              "\u20ac\tR: 100`, group: &quot;states&quot;, x: -56, y: 102 },\n",
-              "{ id: 4, label: `fail test\n",
-              "\u20ac\tR: 0`, group: &quot;states&quot;, x: 58, y: 61 },\n",
-              "{ id: 5, label: `end\n",
-              "\u20ac\tR: 0`, group: &quot;states&quot;, x: 24, y: 187 },\n",
-              "{ id: 10000000000, label: `study\n",
-              "\u20ac\tR: 0`, group: &quot;actions&quot;, x: -136, y: -127 },\n",
-              "{ id: 10000000001, label: `don&#x27;t study\n",
-              "\u20ac\tR: 0`, group: &quot;actions&quot;, x: 77, y: -144 },\n",
-              "]);\n",
-              "        var edges = new vis.DataSet([{ from: 0, to: 10000000000 },\n",
-              "{ from: 10000000000, to: 1, label: &quot;1&quot; },\n",
-              "{ from: 0, to: 10000000001 },\n",
-              "{ from: 10000000001, to: 2, label: &quot;1&quot; },\n",
-              "{ from: 1, to: 3, label: &quot;9/10&quot; },\n",
-              "{ from: 1, to: 4, label: &quot;1/10&quot; },\n",
-              "{ from: 2, to: 3, label: &quot;2/5&quot; },\n",
-              "{ from: 2, to: 4, label: &quot;3/5&quot; },\n",
-              "{ from: 3, to: 5, label: &quot;1&quot; },\n",
-              "{ from: 4, to: 5, label: &quot;1&quot; },\n",
-              "]);\n",
-              "        var options = {\n",
-              "  &quot;__fake_macros&quot;: {\n",
-              "    &quot;__group_macro&quot;: {\n",
-              "      &quot;borderWidth&quot;: 1,\n",
-              "      &quot;color&quot;: {\n",
-              "        &quot;background&quot;: &quot;white&quot;,\n",
-              "        &quot;border&quot;: &quot;black&quot;,\n",
-              "        &quot;highlight&quot;: {\n",
-              "          &quot;background&quot;: &quot;white&quot;,\n",
-              "          &quot;border&quot;: &quot;red&quot;\n",
-              "        }\n",
-              "      },\n",
-              "      &quot;shape&quot;: &quot;ellipse&quot;,\n",
-              "      &quot;mass&quot;: 1,\n",
-              "      &quot;font&quot;: {\n",
-              "        &quot;color&quot;: &quot;black&quot;,\n",
-              "        &quot;size&quot;: 14\n",
-              "      }\n",
-              "    }\n",
-              "  },\n",
-              "  &quot;groups&quot;: {\n",
-              "    &quot;states&quot;: {\n",
-              "      &quot;borderWidth&quot;: 1,\n",
-              "      &quot;color&quot;: {\n",
-              "        &quot;background&quot;: &quot;#8ff0a4&quot;,\n",
-              "        &quot;border&quot;: &quot;black&quot;,\n",
-              "        &quot;highlight&quot;: {\n",
-              "          &quot;background&quot;: &quot;white&quot;,\n",
-              "          &quot;border&quot;: &quot;red&quot;\n",
-              "        }\n",
-              "      },\n",
-              "      &quot;shape&quot;: &quot;ellipse&quot;,\n",
-              "      &quot;mass&quot;: 1,\n",
-              "      &quot;font&quot;: {\n",
-              "        &quot;color&quot;: &quot;black&quot;,\n",
-              "        &quot;size&quot;: 14\n",
-              "      }\n",
-              "    },\n",
-              "    &quot;actions&quot;: {\n",
-              "      &quot;borderWidth&quot;: 1,\n",
-              "      &quot;color&quot;: {\n",
-              "        &quot;background&quot;: &quot;#ffc0cb&quot;,\n",
-              "        &quot;border&quot;: &quot;black&quot;,\n",
-              "        &quot;highlight&quot;: {\n",
-              "          &quot;background&quot;: &quot;white&quot;,\n",
-              "          &quot;border&quot;: &quot;red&quot;\n",
-              "        }\n",
-              "      },\n",
-              "      &quot;shape&quot;: &quot;box&quot;,\n",
-              "      &quot;mass&quot;: 1,\n",
-              "      &quot;font&quot;: {\n",
-              "        &quot;color&quot;: &quot;black&quot;,\n",
-              "        &quot;size&quot;: 14\n",
-              "      }\n",
-              "    },\n",
-              "    &quot;scheduled_actions&quot;: {\n",
-              "      &quot;borderWidth&quot;: 1,\n",
-              "      &quot;color&quot;: {\n",
-              "        &quot;background&quot;: &quot;pink&quot;,\n",
-              "        &quot;border&quot;: &quot;black&quot;,\n",
-              "        &quot;highlight&quot;: {\n",
-              "          &quot;background&quot;: &quot;white&quot;,\n",
-              "          &quot;border&quot;: &quot;red&quot;\n",
-              "        }\n",
-              "      },\n",
-              "      &quot;shape&quot;: &quot;box&quot;,\n",
-              "      &quot;mass&quot;: 1,\n",
-              "      &quot;font&quot;: {\n",
-              "        &quot;color&quot;: &quot;black&quot;,\n",
-              "        &quot;size&quot;: 14\n",
-              "      },\n",
-              "      &quot;schedColor&quot;: false\n",
-              "    }\n",
-              "  },\n",
-              "  &quot;reload_button&quot;: false,\n",
-              "  &quot;edges&quot;: {\n",
-              "    &quot;arrows&quot;: &quot;to&quot;,\n",
-              "    &quot;font&quot;: {\n",
-              "      &quot;color&quot;: &quot;black&quot;,\n",
-              "      &quot;size&quot;: 14\n",
-              "    },\n",
-              "    &quot;color&quot;: {\n",
-              "      &quot;color&quot;: &quot;black&quot;\n",
-              "    }\n",
-              "  },\n",
-              "  &quot;numbers&quot;: {\n",
-              "    &quot;fractions&quot;: true,\n",
-              "    &quot;digits&quot;: 5\n",
-              "  },\n",
-              "  &quot;state_properties&quot;: {\n",
-              "    &quot;show_results&quot;: true,\n",
-              "    &quot;result_symbol&quot;: &quot;\\u2606&quot;,\n",
-              "    &quot;show_rewards&quot;: true,\n",
-              "    &quot;reward_symbol&quot;: &quot;\\u20ac&quot;,\n",
-              "    &quot;show_zero_rewards&quot;: true,\n",
-              "    &quot;show_observations&quot;: true,\n",
-              "    &quot;observation_symbol&quot;: &quot;\\u0298&quot;\n",
-              "  },\n",
-              "  &quot;layout&quot;: {\n",
-              "    &quot;randomSeed&quot;: 5\n",
-              "  },\n",
-              "  &quot;misc&quot;: {\n",
-              "    &quot;enable_physics&quot;: true,\n",
-              "    &quot;width&quot;: 800,\n",
-              "    &quot;height&quot;: 600,\n",
-              "    &quot;explore&quot;: false\n",
-              "  },\n",
-              "  &quot;saving&quot;: {\n",
-              "    &quot;relative_path&quot;: true,\n",
-              "    &quot;filename&quot;: &quot;layouts/pinkgreen.json&quot;,\n",
-              "    &quot;save_button&quot;: false,\n",
-              "    &quot;load_button&quot;: false\n",
-              "  },\n",
-              "  &quot;positions&quot;: {\n",
-              "    &quot;0&quot;: {\n",
-              "      &quot;x&quot;: -27,\n",
-              "      &quot;y&quot;: -203\n",
-              "    },\n",
-              "    &quot;1&quot;: {\n",
-              "      &quot;x&quot;: -84,\n",
-              "      &quot;y&quot;: -6\n",
-              "    },\n",
-              "    &quot;2&quot;: {\n",
-              "      &quot;x&quot;: 57,\n",
-              "      &quot;y&quot;: -10\n",
-              "    },\n",
-              "    &quot;3&quot;: {\n",
-              "      &quot;x&quot;: -56,\n",
-              "      &quot;y&quot;: 102\n",
-              "    },\n",
-              "    &quot;4&quot;: {\n",
-              "      &quot;x&quot;: 58,\n",
-              "      &quot;y&quot;: 61\n",
-              "    },\n",
-              "    &quot;5&quot;: {\n",
-              "      &quot;x&quot;: 24,\n",
-              "      &quot;y&quot;: 187\n",
-              "    },\n",
-              "    &quot;10000000000&quot;: {\n",
-              "      &quot;x&quot;: -136,\n",
-              "      &quot;y&quot;: -127\n",
-              "    },\n",
-              "    &quot;10000000001&quot;: {\n",
-              "      &quot;x&quot;: 77,\n",
-              "      &quot;y&quot;: -144\n",
+              "<script>\n",
+              "function return_id_result(url, id, data) {\n",
+              "        fetch(url, {\n",
+              "            method: 'POST',\n",
+              "            body: JSON.stringify({\n",
+              "                'id': id,\n",
+              "                'data': data\n",
+              "            })\n",
+              "        })\n",
               "    }\n",
-              "  },\n",
-              "  &quot;width&quot;: 800,\n",
-              "  &quot;height&quot;: 600,\n",
-              "  &quot;physics&quot;: true,\n",
-              "  &quot;results_and_rewards&quot;: {\n",
-              "    &quot;show_results&quot;: true,\n",
-              "    &quot;resultSymbol&quot;: &quot;\\u2606&quot;,\n",
-              "    &quot;show_rewards&quot;: true\n",
-              "  }\n",
-              "};\n",
-              "        //js\n",
-              "var container = document.getElementById(&quot;mynetwork&quot;);\n",
-              "var data = {\n",
-              "    nodes: nodes,\n",
-              "    edges: edges,\n",
-              "};\n",
-              "var network = new vis.Network(container, data, options);\n",
-              "function makeAllNodesInvisible() {\n",
-              "    ids = nodes.getIds();\n",
-              "    for (let i = 0; i &lt; ids.length; i++) {\n",
-              "        var nodeId = ids[i];\n",
-              "        var node = nodes.get(nodeId);\n",
-              "        node[&quot;hidden&quot;] = true;\n",
-              "        nodes.update(node);\n",
-              "    }\n",
-              "};\n",
-              "function makeNeighborsVisible(myNode) {\n",
-              "    var ids = network.getConnectedNodes(myNode, &#x27;to&#x27;);\n",
-              "    for (let i = 0; i &lt; ids.length; i++) {\n",
-              "        var nodeId = ids[i];\n",
-              "        var node = nodes.get(nodeId);\n",
-              "        node[&quot;hidden&quot;] = false;\n",
-              "        nodes.update(node);\n",
-              "    }\n",
-              "};\n",
-              "function makeNodeVisible(nodeId) {\n",
-              "    var node = nodes.get(nodeId);\n",
-              "    node[&quot;hidden&quot;] = false;\n",
-              "    nodes.update(node);\n",
-              "};\n",
-              "network.on( &#x27;click&#x27;, function(properties) {\n",
-              "    var nodeId = network.getNodeAt({x:properties.event.srcEvent.offsetX, y:properties.event.srcEvent.offsetY});\n",
-              "    makeNeighborsVisible(nodeId);\n",
-              "});\n",
-              "\n",
-              "    &lt;/script&gt;\n",
-              "  &lt;/body&gt;\n",
-              "&lt;/html&gt;\n",
-              "\"\n",
-              "                border:none !important;\n",
-              "                allowfullscreen webkitallowfullscreen mozallowfullscreen\n",
-              "          ></iframe>"
+              "</script>"
             ],
             "text/plain": [
               "<IPython.core.display.HTML object>"
@@ -425,15 +196,84 @@
           },
           "metadata": {},
           "output_type": "display_data"
+        },
+        {
+          "data": {
+            "application/javascript": [
+              "return_id_result('http://127.0.0.1:8889', 'zZtxQrYJanXwlIbrckDc', 'test message')"
+            ],
+            "text/plain": [
+              "<IPython.core.display.Javascript object>"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        },
+        {
+          "data": {
+            "application/vnd.jupyter.widget-view+json": {
+              "model_id": "f073135af213448c994b303e53484519",
+              "version_major": 2,
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Output()"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        },
+        {
+          "data": {
+            "application/vnd.jupyter.widget-view+json": {
+              "model_id": "bd4476310d4241688ea7f374c549b0b5",
+              "version_major": 2,
+              "version_minor": 0
+            },
+            "text/plain": [
+              "HBox(children=(Output(), Output()))"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        },
+        {
+          "data": {
+            "application/javascript": [
+              "//js\n",
+              "document.getElementById('studywjFNvmgEOX').contentWindow.network.on( 'click', function(properties) {\n",
+              "    var nodeId = document.getElementById('studywjFNvmgEOX').contentWindow.network.getNodeAt({x:properties.event.srcEvent.offsetX, y:properties.event.srcEvent.offsetY});\n",
+              "    if (! (nodeId === undefined)) {\n",
+              "        return_id_result('http://127.0.0.1:8889', 'MoudfJVIPcWgNAgFZTZW', nodeId);\n",
+              "    }\n",
+              "});"
+            ],
+            "text/plain": [
+              "<IPython.core.display.Javascript object>"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        },
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "explore 10000000001\n",
+            "2\n",
+            "make visible 2 <class 'int'>\n",
+            "document.getElementById('studywjFNvmgEOX').contentWindow.nodes.update({ id: 2, label: `didn't study\n",
+            "\u20ac\tR: 15`, group: \"states\", x: 57, y: -10 });\n"
+          ]
         }
       ],
       "source": [
-        "vis = show(mdp, layout=Layout(\"layouts/pinkgreen.json\"), name=\"study\", show_editor=False)"
+        "vis = show(mdp, layout=Layout(\"layouts/pinkgreen.json\"), name=\"study\", show_editor=True)"
       ]
     },
     {
       "cell_type": "code",
-      "execution_count": 10,
+      "execution_count": 9,
       "id": "6cb412a1-52fa-4663-b7ba-371e06001f7a",
       "metadata": {},
       "outputs": [
@@ -441,19 +281,86 @@
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "{0: {Edge(from_=0, to_=10000000001, label=None)}, 10000000000: set(), 10000000001: set(), 1: {Edge(from_=1, to_=4, label='1/10')}, 2: {Edge(from_=2, to_=4, label='3/5')}, 3: set(), 4: set()}\n"
+            "{0: {Edge(from_=0, to_=10000000000, label=None), Edge(from_=0, to_=10000000001, label=None)}, 10000000000: {Edge(from_=10000000000, to_=1, label='1')}, 10000000001: {Edge(from_=10000000001, to_=2, label='1')}, 1: {Edge(from_=1, to_=3, label='9/10'), Edge(from_=1, to_=4, label='1/10')}, 2: {Edge(from_=2, to_=4, label='3/5'), Edge(from_=2, to_=3, label='2/5')}, 3: {Edge(from_=3, to_=5, label='1')}, 4: {Edge(from_=4, to_=5, label='1')}}\n"
           ]
         }
       ],
       "source": [
-        "\n",
         "print(vis.nt.edges)"
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": 10,
+      "id": "f281c503-d45e-4f36-a368-71bda9e0a898",
+      "metadata": {
+        "scrolled": true
+      },
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "make visible 10000000001 <class 'int'>\n",
+            "document.getElementById('studywjFNvmgEOX').contentWindow.nodes.update({ id: 10000000001, label: `don't study\n",
+            "\u20ac\tR: 0`, group: \"actions\", x: 77, y: -144 });\n"
+          ]
+        },
+        {
+          "data": {
+            "application/javascript": [
+              "document.getElementById('studywjFNvmgEOX').contentWindow.nodes.update({ id: 10000000001, label: `don't study\n",
+              "\u20ac\tR: 0`, group: \"actions\", x: 77, y: -144 });"
+            ],
+            "text/plain": [
+              "<IPython.core.display.Javascript object>"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        }
+      ],
+      "source": [
+        "vis.nt.make_node_visible(10000000001, True)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 11,
+      "id": "fee61f37-2e8e-48d0-9d57-e7f92c3effab",
+      "metadata": {},
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "make visible 2 <class 'int'>\n",
+            "document.getElementById('studywjFNvmgEOX').contentWindow.nodes.update({ id: 2, label: `didn't study\n",
+            "\u20ac\tR: 15`, group: \"states\", x: 57, y: -10 });\n"
+          ]
+        },
+        {
+          "data": {
+            "application/javascript": [
+              "document.getElementById('studywjFNvmgEOX').contentWindow.nodes.update({ id: 2, label: `didn't study\n",
+              "\u20ac\tR: 15`, group: \"states\", x: 57, y: -10 });"
+            ],
+            "text/plain": [
+              "<IPython.core.display.Javascript object>"
+            ]
+          },
+          "metadata": {},
+          "output_type": "display_data"
+        }
+      ],
+      "source": [
+        "vis.nt.make_node_visible(2, True)"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": null,
-      "id": "f1a87851-75b5-4e49-91ab-9382f711e495",
+      "id": "5ffb652e-e2ca-4212-b8b3-44db33ba46f3",
       "metadata": {},
       "outputs": [],
       "source": []
diff --git a/docs/getting_started/layouts/pinkgreen.json b/docs/getting_started/layouts/pinkgreen.json
index 799d8a7..5b72941 100644
--- a/docs/getting_started/layouts/pinkgreen.json
+++ b/docs/getting_started/layouts/pinkgreen.json
@@ -103,7 +103,7 @@
     "enable_physics": true,
     "width": 800,
     "height": 600,
-    "explore": false
+    "explore": true
   },
   "saving": {
     "relative_path": true,
diff --git a/docs/getting_started/study.html b/docs/getting_started/study.html
index 2e637d8..2228f93 100644
--- a/docs/getting_started/study.html
+++ b/docs/getting_started/study.html
@@ -1,6 +1,6 @@
 
           <iframe
-                id="studysFvnWfhilS"
+                id="studymrkLMSGkNb"
                 width="820"
                 height="620"
                 frameborder="0"
@@ -217,33 +217,6 @@
     edges: edges,
 };
 var network = new vis.Network(container, data, options);
-function makeAllNodesInvisible() {
-    ids = nodes.getIds();
-    for (let i = 0; i &lt; ids.length; i++) {
-        var nodeId = ids[i];
-        var node = nodes.get(nodeId);
-        node[&quot;hidden&quot;] = true;
-        nodes.update(node);
-    }
-};
-function makeNeighborsVisible(myNode) {
-    var ids = network.getConnectedNodes(myNode, &#x27;to&#x27;);
-    for (let i = 0; i &lt; ids.length; i++) {
-        var nodeId = ids[i];
-        var node = nodes.get(nodeId);
-        node[&quot;hidden&quot;] = false;
-        nodes.update(node);
-    }
-};
-function makeNodeVisible(nodeId) {
-    var node = nodes.get(nodeId);
-    node[&quot;hidden&quot;] = false;
-    nodes.update(node);
-};
-network.on( &#x27;click&#x27;, function(properties) {
-    var nodeId = network.getNodeAt({x:properties.event.srcEvent.offsetX, y:properties.event.srcEvent.offsetY});
-    makeNeighborsVisible(nodeId);
-});
 
     &lt;/script&gt;
   &lt;/body&gt;
diff --git a/stormvogel/html_templates.py b/stormvogel/html_templates.py
index 19a3360..04d58e7 100644
--- a/stormvogel/html_templates.py
+++ b/stormvogel/html_templates.py
@@ -33,6 +33,19 @@
     edges: edges,
 };
 var network = new vis.Network(container, data, options);
+// network.on( 'click', function(properties) {
+//     var nodeId = network.getNodeAt({x:properties.event.srcEvent.offsetX, y:properties.event.srcEvent.offsetY});
+//     alert(nodeId);
+// });
+"""
+
+NETWORK_JS_OLD = """//js
+var container = document.getElementById("mynetwork");
+var data = {
+    nodes: nodes,
+    edges: edges,
+};
+var network = new vis.Network(container, data, options);
 function makeAllNodesInvisible() {
     ids = nodes.getIds();
     for (let i = 0; i < ids.length; i++) {
@@ -56,8 +69,4 @@
     node["hidden"] = false;
     nodes.update(node);
 };
-network.on( 'click', function(properties) {
-    var nodeId = network.getNodeAt({x:properties.event.srcEvent.offsetX, y:properties.event.srcEvent.offsetY});
-    makeNeighborsVisible(nodeId);
-});
 """
diff --git a/stormvogel/show.py b/stormvogel/show.py
index 1907eac..473f1ee 100644
--- a/stormvogel/show.py
+++ b/stormvogel/show.py
@@ -58,6 +58,7 @@ def show(
         e.show()
         box = widgets.HBox(children=[vis.output, e.output])
         ipd.display(box)
+        vis.after_show()
     else:  # Unfortunately, the sphinx docs only work if we save the html as a file and embed.
         iframe = vis.nt.generate_iframe()
         with open(name + ".html", "w") as f:
diff --git a/stormvogel/visjs.py b/stormvogel/visjs.py
index 56c1677..b13d776 100644
--- a/stormvogel/visjs.py
+++ b/stormvogel/visjs.py
@@ -16,7 +16,7 @@
 spam: widgets.Output = widgets.Output()
 
 
-@dataclass(frozen=True)
+@dataclass
 class Node:
     id: int
     label: str | None
@@ -76,15 +76,40 @@ def __init__(
             self.positions = positions
         self.nodes: dict[int, Node] = {}
         self.edges: dict[int, set[Edge]] = {}
+        self.reverse_edges: dict[int, set[Edge]] = {}
 
     def enable_exploration_mode(self):
-        """Make the initial node visible, and inject the JS that enables exploration by clicking."""
+        """Inject the JS that enables exploration by clicking."""
+        js = f"""//js
+{self.content_window}.network.on( 'click', function(properties) {{
+    var nodeId = {self.content_window}.network.getNodeAt({{x:properties.event.srcEvent.offsetX, y:properties.event.srcEvent.offsetY}});
+    if (! (nodeId === undefined)) {{
+        FUNCTION(nodeId);
+    }}
+}});"""
+        self.server.add_event(js, self.make_neighbors_visible)
+
+    def make_neighbors_visible(self, id: str) -> None:
+        """Make the neighbors of this node visible."""
+        print("explore", id)
+        for edge in self.edges[int(id)]:
+            print(edge.to_)
+            self.make_node_visible(edge.to_, True)
 
-    def make_node_visible(self, id):
-        """Make the node with this id visible."""
+    def make_node_visible(self, id: int, two: bool = False) -> None:
+        """Make the node with this id visible, i.e. adding it to the JS network.
+        Add incoming edges if the source node is visible and outgoing edges if the target node is visible."""
+        print("make visible", id, type(id))
+        node = self.nodes[id]
+        node.visible = True
+        js = f"""alert('{id}');"""
+        if two:
+            js = f"""{self.content_window}.nodes.update({self.__create_node_code(node)});"""
+        print(js)
+        ipd.display(ipd.Javascript(js))
 
     def get_positions(self) -> dict:
-        """Get the current positions of the nodes on the canvas. Returns empty dict if unsucessful.
+        """Get the current positions of the nodes on the canvas. Returns empty dict if unsuccessful.
         Example result: {"0": {"x": 5, "y": 10}}"""
         if self.server is None:
             with self.debug_output:
@@ -104,8 +129,8 @@ def get_positions(self) -> dict:
                 logging.warning("Timed out. Could not retrieve position data.")
             raise TimeoutError("Timed out. Could not retrieve position data.")
 
-    def __add_node_pre(self, node: Node) -> None:
-        """Add the node to the pre-creation Javascript."""
+    def __create_node_code(self, node: Node) -> str:
+        """Generate the code to add the node to the pre-creation Javascript."""
         current = "{ id: " + str(node.id)
         if node.label is not None:
             current += f", label: `{node.label}`"
@@ -113,8 +138,8 @@ def __add_node_pre(self, node: Node) -> None:
             current += f', group: "{node.group}"'
         if self.positions is not None and str(node.id) in self.positions:
             current += f', x: {self.positions[str(node.id)]["x"]}, y: {self.positions[str(node.id)]["y"]}'
-        current += " },\n"
-        self.nodes_js += current
+        current += " }"
+        return current
 
     def add_node(
         self,
@@ -127,15 +152,15 @@ def add_node(
         node = Node(id, label, group, visible)
         self.nodes[id] = node
         if visible:
-            self.__add_node_pre(node)
+            self.nodes_js += self.__create_node_code(node) + ",\n"
 
-    def add_edge_pre(self, edge: Edge):
+    def __create_edge_code(self, edge: Edge) -> str:
         """Add the edge to the pre-creation Javascript."""
         current = "{ from: " + str(edge.from_) + ", to: " + str(edge.to_)
         if edge.label is not None:
             current += f', label: "{edge.label}"'
         current += " },\n"
-        self.edges_js += current
+        return current
 
     def add_edge(
         self,
@@ -148,9 +173,14 @@ def add_edge(
         if from_ in self.edges:
             self.edges[from_].add(edge)
         else:
-            self.edges[from_] = set()
+            self.edges[from_] = {edge}
+        if to_ in self.reverse_edges:
+            self.reverse_edges[to_].add(edge)
+        else:
+            self.reverse_edges[to_] = {edge}
+
         if self.nodes[from_].visible and self.nodes[from_].visible:
-            self.add_edge_pre(edge)
+            self.edges_js += self.__create_edge_code(edge)
 
     def set_options(self, options: str) -> None:
         """Set the options. Only use before calling show."""
diff --git a/stormvogel/visualization.py b/stormvogel/visualization.py
index 49e3938..f961025 100644
--- a/stormvogel/visualization.py
+++ b/stormvogel/visualization.py
@@ -3,6 +3,7 @@
 # Note to future maintainers: The way that IPython display behaves is very flakey sometimes.
 # If you remove a with output: statement, everything might just break, be prepared.
 
+from time import sleep
 import stormvogel.model
 import stormvogel.layout
 import stormvogel.result
@@ -104,9 +105,6 @@ def show(self) -> None:
         with self.output:
             ipd.clear_output()
         self.__create_nt()
-        if self.layout.layout["misc"]["explore"]:
-            self.nt.enable_exploration_mode()
-            self.nt.make_node_visible(self.model.get_initial_state().id)
         self.layout.set_groups(self.separate_labels)
         self.__add_states()
         self.__add_transitions()
@@ -116,6 +114,14 @@ def show(self) -> None:
             self.nt.show()
         self.maybe_display_output()
 
+    def after_show(self) -> None:
+        """Initialization that needs to be called after the network is shown."""
+        sleep(1)  # Delay because JS needs some time to load.
+        if self.layout.layout["misc"]["explore"]:
+            self.nt.enable_exploration_mode()
+            # sleep(3)
+            # self.nt.make_node_visible(self.model.get_initial_state().id, two=True)
+
     def update(self) -> None:
         """Tries to update an existing visualization to apply layout changes WITHOUT reloading. If show was not called before, nothing happens."""
         if self.nt is not None:
@@ -184,6 +190,7 @@ def __add_transitions(self) -> None:
                         id=action_id,
                         label=",".join(action.labels) + reward,
                         group=group,
+                        visible=not self.layout.layout["misc"]["explore"],
                     )
                     # Add transition from this state TO the action.
                     self.nt.add_edge(state_id, action_id)  # type: ignore