diff --git a/docs/getting_started/01_study.ipynb b/docs/getting_started/01_study.ipynb
index 23173e4..5cf41b8 100644
--- a/docs/getting_started/01_study.ipynb
+++ b/docs/getting_started/01_study.ipynb
@@ -155,7 +155,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 8,
+      "execution_count": 14,
       "id": "0df1511e-565d-45d0-93a8-adafbfaaaefa",
       "metadata": {
         "scrolled": true
@@ -164,7 +164,7 @@
         {
           "data": {
             "application/vnd.jupyter.widget-view+json": {
-              "model_id": "be7c1a38bc504a67abc215743d681813",
+              "model_id": "bd157ce6c3034fd5a0533f8220e44c9e",
               "version_major": 2,
               "version_minor": 0
             },
@@ -175,44 +175,10 @@
           "metadata": {},
           "output_type": "display_data"
         },
-        {
-          "data": {
-            "text/html": [
-              "<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",
-              "</script>"
-            ],
-            "text/plain": [
-              "<IPython.core.display.HTML object>"
-            ]
-          },
-          "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",
+              "model_id": "f7080afa9cce47deb47ef9c1b145ce71",
               "version_major": 2,
               "version_minor": 0
             },
@@ -226,7 +192,7 @@
         {
           "data": {
             "application/vnd.jupyter.widget-view+json": {
-              "model_id": "bd4476310d4241688ea7f374c549b0b5",
+              "model_id": "69b3da5b0e884431b9cb84d758067af4",
               "version_major": 2,
               "version_minor": 0
             },
@@ -236,131 +202,16 @@
           },
           "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=True)"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 9,
-      "id": "6cb412a1-52fa-4663-b7ba-371e06001f7a",
-      "metadata": {},
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "{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": [
-        "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)"
+        "vis = show(mdp, layout=Layout(\"layouts/pinkgreen.json\"), name=\"study\", show_editor=False)"
       ]
     },
     {
       "cell_type": "code",
       "execution_count": null,
-      "id": "5ffb652e-e2ca-4212-b8b3-44db33ba46f3",
+      "id": "a9e68b97-49d8-47dc-8ba1-5d179bcecc52",
       "metadata": {},
       "outputs": [],
       "source": []
diff --git a/docs/getting_started/layouts/pinkgreen.json b/docs/getting_started/layouts/pinkgreen.json
index 5b72941..799d8a7 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": true
+    "explore": false
   },
   "saving": {
     "relative_path": true,
diff --git a/docs/getting_started/study.html b/docs/getting_started/study.html
index 2228f93..e7edec5 100644
--- a/docs/getting_started/study.html
+++ b/docs/getting_started/study.html
@@ -1,6 +1,6 @@
 
           <iframe
-                id="studymrkLMSGkNb"
+                id="studyNsXSOjuhzk"
                 width="820"
                 height="620"
                 frameborder="0"
@@ -217,6 +217,33 @@
     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 04d58e7..19a3360 100644
--- a/stormvogel/html_templates.py
+++ b/stormvogel/html_templates.py
@@ -33,19 +33,6 @@
     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++) {
@@ -69,4 +56,8 @@
     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 473f1ee..1907eac 100644
--- a/stormvogel/show.py
+++ b/stormvogel/show.py
@@ -58,7 +58,6 @@ 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 b13d776..01d7e74 100644
--- a/stormvogel/visjs.py
+++ b/stormvogel/visjs.py
@@ -10,27 +10,11 @@
 import random
 import string
 import logging
-from dataclasses import dataclass
 
 
 spam: widgets.Output = widgets.Output()
 
 
-@dataclass
-class Node:
-    id: int
-    label: str | None
-    group: str | None
-    visible: bool
-
-
-@dataclass(frozen=True)
-class Edge:
-    from_: int
-    to_: int
-    label: str | None
-
-
 class Network(stormvogel.displayable.Displayable):
     EXTRA_PIXELS: int = 20  # To prevent the scroll bar around the Network.
 
@@ -65,6 +49,7 @@ def __init__(
         self.nodes_js: str = ""
         self.edges_js: str = ""
         self.options_js: str = "{}"
+        self.new_nodes_hidden: bool = False
         if do_init_server:
             self.server: stormvogel.communication_server.CommunicationServer = (
                 stormvogel.communication_server.initialize_server()
@@ -74,42 +59,24 @@ def __init__(
             self.positions = {}
         else:
             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):
-        """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: 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))
+        # Note that this refers to the same server as the global variable in stormvogel.communication_server.
+
+    def enable_exploration_mode(self, initial_node_id: int):
+        """Every node becomes invisible. You can then click any node to reveal all of its successors. Call before adding any nodes to the network."""
+        self.new_nodes_hidden = True
+        self.initial_node_id = initial_node_id
+
+    def update_exploration_mode(self, initial_node_id: int):
+        # Make all nodes invisible.
+        ipd.display(ipd.Javascript(self.content_window + ".makeAllNodesInvisible()"))
+        # Make the initial state visible.
+        ipd.display(
+            ipd.Javascript(self.content_window + f".makeNodeVisible({initial_node_id})")
+        )
+        # All future nodes to be added will be hidden as well.
 
     def get_positions(self) -> dict:
-        """Get the current positions of the nodes on the canvas. Returns empty dict if unsuccessful.
+        """Get the current positions of the nodes on the canvas. Returns empty dict if unsucessful.
         Example result: {"0": {"x": 5, "y": 10}}"""
         if self.server is None:
             with self.debug_output:
@@ -129,58 +96,37 @@ def get_positions(self) -> dict:
                 logging.warning("Timed out. Could not retrieve position data.")
             raise TimeoutError("Timed out. Could not retrieve position data.")
 
-    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}`"
-        if node.group is not 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 += " }"
-        return current
-
     def add_node(
         self,
         id: int,
         label: str | None = None,
         group: str | None = None,
-        visible: bool = True,
     ) -> None:
         """Add a node. Only use before calling show."""
-        node = Node(id, label, group, visible)
-        self.nodes[id] = node
-        if visible:
-            self.nodes_js += self.__create_node_code(node) + ",\n"
-
-    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 = "{ id: " + str(id)
+        if label is not None:
+            current += f", label: `{label}`"
+        if group is not None:
+            current += f', group: "{group}"'
+        if self.positions is not None and str(id) in self.positions:
+            current += f', x: {self.positions[str(id)]["x"]}, y: {self.positions[str(id)]["y"]}'
+        if self.new_nodes_hidden and id != self.initial_node_id:
+            current += ", hidden: true"
         current += " },\n"
-        return current
+        self.nodes_js += current
 
     def add_edge(
         self,
         from_: int,
-        to_: int,
+        to: int,
         label: str | None = None,
     ) -> None:
         """Add an edge. Only use before calling show."""
-        edge = Edge(from_, to_, label)
-        if from_ in self.edges:
-            self.edges[from_].add(edge)
-        else:
-            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.edges_js += self.__create_edge_code(edge)
+        current = "{ from: " + str(from_) + ", to: " + str(to)
+        if label is not None:
+            current += f', label: "{label}"'
+        current += " },\n"
+        self.edges_js += current
 
     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 f961025..42f24e5 100644
--- a/stormvogel/visualization.py
+++ b/stormvogel/visualization.py
@@ -3,7 +3,6 @@
 # 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
@@ -105,6 +104,8 @@ 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.model.get_initial_state().id)
         self.layout.set_groups(self.separate_labels)
         self.__add_states()
         self.__add_transitions()
@@ -114,14 +115,6 @@ 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:
@@ -150,7 +143,6 @@ def __add_states(self) -> None:
                 state.id,
                 label=",".join(state.labels) + rewards + res + observations,
                 group=group,
-                visible=not self.layout.layout["misc"]["explore"],
             )
 
     def __add_transitions(self) -> None:
@@ -190,7 +182,6 @@ 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