From a000256a82826065325c8a187b8204ae302cb3e4 Mon Sep 17 00:00:00 2001
From: Ivo Melse <ivo@melse.info>
Date: Wed, 26 Jun 2024 13:20:03 +0200
Subject: [PATCH] Start with visualization

---
 notebooks/die.ipynb         |  90 ++++++++-
 notebooks/diegraph.html     | 321 ++++++++++++++++++++++++++++++++
 notebooks/hi.html           | 361 ++++++++++++++++++++++++++++++++++++
 notebooks/monty.html        | 321 ++++++++++++++++++++++++++++++++
 notebooks/monty_hall.ipynb  | 131 +++++++++++++
 stormvogel/visualization.py |  76 +++++++-
 6 files changed, 1286 insertions(+), 14 deletions(-)
 create mode 100644 notebooks/diegraph.html
 create mode 100644 notebooks/hi.html
 create mode 100644 notebooks/monty.html
 create mode 100644 notebooks/monty_hall.ipynb

diff --git a/notebooks/die.ipynb b/notebooks/die.ipynb
index 9822be1..7137aad 100644
--- a/notebooks/die.ipynb
+++ b/notebooks/die.ipynb
@@ -6,41 +6,111 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from stormvogel.visualization import make_slider"
+    "import stormvogel.model\n",
+    "import stormvogel.visualization"
    ]
   },
   {
    "cell_type": "code",
    "execution_count": 2,
    "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Create a new model with the name \"Die\"\n",
+    "dtmc = stormvogel.model.new_dtmc(\"Die\")\n",
+    "\n",
+    "init = dtmc.get_initial_state()\n",
+    "\n",
+    "# From the initial state, add the transition to 6 new states with probability 1/6th.\n",
+    "init.set_transitions(\n",
+    "    [(1 / 6, dtmc.new_state(f\"rolled{i}\", {\"rolled\": i})) for i in range(6)]\n",
+    ")\n",
+    "\n",
+    "# Print the resulting model in dot format.\n",
+    "# print(dtmc.to_dot())"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {
+    "scrolled": true
+   },
    "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "diegraph.html\n"
+     ]
+    },
     {
      "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "9caed9de44144d50b26a7e76ef0accb8",
-       "version_major": 2,
-       "version_minor": 0
-      },
+      "text/html": [
+       "\n",
+       "        <iframe\n",
+       "            width=\"100%\"\n",
+       "            height=\"600px\"\n",
+       "            src=\"diegraph.html\"\n",
+       "            frameborder=\"0\"\n",
+       "            allowfullscreen\n",
+       "            \n",
+       "        ></iframe>\n",
+       "        "
+      ],
       "text/plain": [
-       "interactive(children=(IntSlider(value=10, description='x', max=30, min=-10), Output()), _dom_classes=('widget-…"
+       "<IPython.lib.display.IFrame at 0x7ff90d17b510>"
       ]
      },
      "metadata": {},
      "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "stormvogel.visualization.show(model=dtmc, name=\"diegraph\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "hi.html\n"
+     ]
     },
     {
      "data": {
+      "text/html": [
+       "\n",
+       "        <iframe\n",
+       "            width=\"100%\"\n",
+       "            height=\"600px\"\n",
+       "            src=\"hi.html\"\n",
+       "            frameborder=\"0\"\n",
+       "            allowfullscreen\n",
+       "            \n",
+       "        ></iframe>\n",
+       "        "
+      ],
       "text/plain": [
-       "<function stormvogel.visualization.make_slider.<locals>.<lambda>(x)>"
+       "<IPython.lib.display.IFrame at 0x7ff90ec3a110>"
       ]
      },
-     "execution_count": 2,
+     "execution_count": 4,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "make_slider()"
+    "from pyvis.network import Network\n",
+    "\n",
+    "g = Network(notebook=True,cdn_resources=\"remote\")\n",
+    "g.show_buttons()\n",
+    "g.show(\"hi.html\")"
    ]
   },
   {
diff --git a/notebooks/diegraph.html b/notebooks/diegraph.html
new file mode 100644
index 0000000..c317554
--- /dev/null
+++ b/notebooks/diegraph.html
@@ -0,0 +1,321 @@
+<html>
+    <head>
+        <meta charset="utf-8">
+        
+            <script>function neighbourhoodHighlight(params) {
+  // console.log("in nieghbourhoodhighlight");
+  allNodes = nodes.get({ returnType: "Object" });
+  // originalNodes = JSON.parse(JSON.stringify(allNodes));
+  // if something is selected:
+  if (params.nodes.length > 0) {
+    highlightActive = true;
+    var i, j;
+    var selectedNode = params.nodes[0];
+    var degrees = 2;
+
+    // mark all nodes as hard to read.
+    for (let nodeId in allNodes) {
+      // nodeColors[nodeId] = allNodes[nodeId].color;
+      allNodes[nodeId].color = "rgba(200,200,200,0.5)";
+      if (allNodes[nodeId].hiddenLabel === undefined) {
+        allNodes[nodeId].hiddenLabel = allNodes[nodeId].label;
+        allNodes[nodeId].label = undefined;
+      }
+    }
+    var connectedNodes = network.getConnectedNodes(selectedNode);
+    var allConnectedNodes = [];
+
+    // get the second degree nodes
+    for (i = 1; i < degrees; i++) {
+      for (j = 0; j < connectedNodes.length; j++) {
+        allConnectedNodes = allConnectedNodes.concat(
+          network.getConnectedNodes(connectedNodes[j])
+        );
+      }
+    }
+
+    // all second degree nodes get a different color and their label back
+    for (i = 0; i < allConnectedNodes.length; i++) {
+      // allNodes[allConnectedNodes[i]].color = "pink";
+      allNodes[allConnectedNodes[i]].color = "rgba(150,150,150,0.75)";
+      if (allNodes[allConnectedNodes[i]].hiddenLabel !== undefined) {
+        allNodes[allConnectedNodes[i]].label =
+          allNodes[allConnectedNodes[i]].hiddenLabel;
+        allNodes[allConnectedNodes[i]].hiddenLabel = undefined;
+      }
+    }
+
+    // all first degree nodes get their own color and their label back
+    for (i = 0; i < connectedNodes.length; i++) {
+      // allNodes[connectedNodes[i]].color = undefined;
+      allNodes[connectedNodes[i]].color = nodeColors[connectedNodes[i]];
+      if (allNodes[connectedNodes[i]].hiddenLabel !== undefined) {
+        allNodes[connectedNodes[i]].label =
+          allNodes[connectedNodes[i]].hiddenLabel;
+        allNodes[connectedNodes[i]].hiddenLabel = undefined;
+      }
+    }
+
+    // the main node gets its own color and its label back.
+    // allNodes[selectedNode].color = undefined;
+    allNodes[selectedNode].color = nodeColors[selectedNode];
+    if (allNodes[selectedNode].hiddenLabel !== undefined) {
+      allNodes[selectedNode].label = allNodes[selectedNode].hiddenLabel;
+      allNodes[selectedNode].hiddenLabel = undefined;
+    }
+  } else if (highlightActive === true) {
+    // console.log("highlightActive was true");
+    // reset all nodes
+    for (let nodeId in allNodes) {
+      // allNodes[nodeId].color = "purple";
+      allNodes[nodeId].color = nodeColors[nodeId];
+      // delete allNodes[nodeId].color;
+      if (allNodes[nodeId].hiddenLabel !== undefined) {
+        allNodes[nodeId].label = allNodes[nodeId].hiddenLabel;
+        allNodes[nodeId].hiddenLabel = undefined;
+      }
+    }
+    highlightActive = false;
+  }
+
+  // transform the object into an array
+  var updateArray = [];
+  if (params.nodes.length > 0) {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        // console.log(allNodes[nodeId]);
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  } else {
+    // console.log("Nothing was selected");
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        // console.log(allNodes[nodeId]);
+        // allNodes[nodeId].color = {};
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  }
+}
+
+function filterHighlight(params) {
+  allNodes = nodes.get({ returnType: "Object" });
+  // if something is selected:
+  if (params.nodes.length > 0) {
+    filterActive = true;
+    let selectedNodes = params.nodes;
+
+    // hiding all nodes and saving the label
+    for (let nodeId in allNodes) {
+      allNodes[nodeId].hidden = true;
+      if (allNodes[nodeId].savedLabel === undefined) {
+        allNodes[nodeId].savedLabel = allNodes[nodeId].label;
+        allNodes[nodeId].label = undefined;
+      }
+    }
+
+    for (let i=0; i < selectedNodes.length; i++) {
+      allNodes[selectedNodes[i]].hidden = false;
+      if (allNodes[selectedNodes[i]].savedLabel !== undefined) {
+        allNodes[selectedNodes[i]].label = allNodes[selectedNodes[i]].savedLabel;
+        allNodes[selectedNodes[i]].savedLabel = undefined;
+      }
+    }
+
+  } else if (filterActive === true) {
+    // reset all nodes
+    for (let nodeId in allNodes) {
+      allNodes[nodeId].hidden = false;
+      if (allNodes[nodeId].savedLabel !== undefined) {
+        allNodes[nodeId].label = allNodes[nodeId].savedLabel;
+        allNodes[nodeId].savedLabel = undefined;
+      }
+    }
+    filterActive = false;
+  }
+
+  // transform the object into an array
+  var updateArray = [];
+  if (params.nodes.length > 0) {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  } else {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  }
+}
+
+function selectNode(nodes) {
+  network.selectNodes(nodes);
+  neighbourhoodHighlight({ nodes: nodes });
+  return nodes;
+}
+
+function selectNodes(nodes) {
+  network.selectNodes(nodes);
+  filterHighlight({nodes: nodes});
+  return nodes;
+}
+
+function highlightFilter(filter) {
+  let selectedNodes = []
+  let selectedProp = filter['property']
+  if (filter['item'] === 'node') {
+    let allNodes = nodes.get({ returnType: "Object" });
+    for (let nodeId in allNodes) {
+      if (allNodes[nodeId][selectedProp] && filter['value'].includes((allNodes[nodeId][selectedProp]).toString())) {
+        selectedNodes.push(nodeId)
+      }
+    }
+  }
+  else if (filter['item'] === 'edge'){
+    let allEdges = edges.get({returnType: 'object'});
+    // check if the selected property exists for selected edge and select the nodes connected to the edge
+    for (let edge in allEdges) {
+      if (allEdges[edge][selectedProp] && filter['value'].includes((allEdges[edge][selectedProp]).toString())) {
+        selectedNodes.push(allEdges[edge]['from'])
+        selectedNodes.push(allEdges[edge]['to'])
+      }
+    }
+  }
+  selectNodes(selectedNodes)
+}</script>
+            <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/vis-network/9.1.2/dist/dist/vis-network.min.css" integrity="sha512-WgxfT5LWjfszlPHXRmBWHkV2eceiWTOBvrKCNbdgDYTHrT2AeLCGbF4sZlZw3UMN3WtL0tGUoIAKsu8mllg/XA==" crossorigin="anonymous" referrerpolicy="no-referrer" />
+            <script src="https://cdnjs.cloudflare.com/ajax/libs/vis-network/9.1.2/dist/vis-network.min.js" integrity="sha512-LnvoEWDFrqGHlHmDD2101OrLcbsfkrzoSpvtSQtxK3RMnRV0eOkhhBN2dXHKRrUU8p2DGRTk35n4O8nWSVe1mQ==" crossorigin="anonymous" referrerpolicy="no-referrer"></script>
+            
+            
+            
+            
+            
+            
+
+        
+<center>
+<h1></h1>
+</center>
+
+<!-- <link rel="stylesheet" href="../node_modules/vis/dist/vis.min.css" type="text/css" />
+<script type="text/javascript" src="../node_modules/vis/dist/vis.js"> </script>-->
+        <link
+          href="https://cdn.jsdelivr.net/npm/bootstrap@5.0.0-beta3/dist/css/bootstrap.min.css"
+          rel="stylesheet"
+          integrity="sha384-eOJMYsd53ii+scO/bJGFsiCZc+5NDVN2yr8+0RDqr0Ql0h+rP48ckxlpbzKgwra6"
+          crossorigin="anonymous"
+        />
+        <script
+          src="https://cdn.jsdelivr.net/npm/bootstrap@5.0.0-beta3/dist/js/bootstrap.bundle.min.js"
+          integrity="sha384-JEW9xMcG8R+pH31jmWH6WWP0WintQrMb4s7ZOdauHnUtxwoG2vI5DkLtS3qm9Ekf"
+          crossorigin="anonymous"
+        ></script>
+
+
+        <center>
+          <h1></h1>
+        </center>
+        <style type="text/css">
+
+             #mynetwork {
+                 width: 100%;
+                 height: 600px;
+                 background-color: #ffffff;
+                 border: 1px solid lightgray;
+                 position: relative;
+                 float: left;
+             }
+
+             
+
+             
+
+             
+        </style>
+    </head>
+
+
+    <body>
+        <div class="card" style="width: 100%">
+            
+            
+            <div id="mynetwork" class="card-body"></div>
+        </div>
+
+        
+        
+
+        <script type="text/javascript">
+
+              // initialize global variables.
+              var edges;
+              var nodes;
+              var allNodes;
+              var allEdges;
+              var nodeColors;
+              var originalNodes;
+              var network;
+              var container;
+              var options, data;
+              var filter = {
+                  item : '',
+                  property : '',
+                  value : []
+              };
+
+              
+
+              
+
+              // This method is responsible for drawing the graph, returns the drawn network
+              function drawGraph() {
+                  var container = document.getElementById('mynetwork');
+
+                  
+
+                  // parsing and collecting nodes and edges from the python
+                  nodes = new vis.DataSet([{"borderWidth": 3, "color": null, "id": 0, "label": "init", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 1, "label": "rolled0", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 2, "label": "rolled1", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 3, "label": "rolled2", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 4, "label": "rolled3", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 5, "label": "rolled4", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 6, "label": "rolled5", "shape": "dot"}]);
+                  edges = new vis.DataSet([{"arrows": "to", "color": "red", "from": 0, "label": "1/6", "to": 1}, {"arrows": "to", "color": "red", "from": 0, "label": "1/6", "to": 2}, {"arrows": "to", "color": "red", "from": 0, "label": "1/6", "to": 3}, {"arrows": "to", "color": "red", "from": 0, "label": "1/6", "to": 4}, {"arrows": "to", "color": "red", "from": 0, "label": "1/6", "to": 5}, {"arrows": "to", "color": "red", "from": 0, "label": "1/6", "to": 6}, {"arrows": "to", "color": "red", "from": 1, "label": "1", "to": 1}, {"arrows": "to", "color": "red", "from": 2, "label": "1", "to": 2}, {"arrows": "to", "color": "red", "from": 3, "label": "1", "to": 3}, {"arrows": "to", "color": "red", "from": 4, "label": "1", "to": 4}, {"arrows": "to", "color": "red", "from": 5, "label": "1", "to": 5}, {"arrows": "to", "color": "red", "from": 6, "label": "1", "to": 6}]);
+
+                  nodeColors = {};
+                  allNodes = nodes.get({ returnType: "Object" });
+                  for (nodeId in allNodes) {
+                    nodeColors[nodeId] = allNodes[nodeId].color;
+                  }
+                  allEdges = edges.get({ returnType: "Object" });
+                  // adding nodes and edges to the graph
+                  data = {nodes: nodes, edges: edges};
+
+                  var options = {"nodes": {"color": {"background": "white", "border": "black"}}, "edge": {"color": "blue"}};
+
+                  
+
+
+                  
+
+                  network = new vis.Network(container, data, options);
+
+                  
+
+                  
+
+                  
+
+
+                  
+
+                  return network;
+
+              }
+              drawGraph();
+        </script>
+    </body>
+</html>
\ No newline at end of file
diff --git a/notebooks/hi.html b/notebooks/hi.html
new file mode 100644
index 0000000..cb4eb42
--- /dev/null
+++ b/notebooks/hi.html
@@ -0,0 +1,361 @@
+<html>
+    <head>
+        <meta charset="utf-8">
+        
+            <script>function neighbourhoodHighlight(params) {
+  // console.log("in nieghbourhoodhighlight");
+  allNodes = nodes.get({ returnType: "Object" });
+  // originalNodes = JSON.parse(JSON.stringify(allNodes));
+  // if something is selected:
+  if (params.nodes.length > 0) {
+    highlightActive = true;
+    var i, j;
+    var selectedNode = params.nodes[0];
+    var degrees = 2;
+
+    // mark all nodes as hard to read.
+    for (let nodeId in allNodes) {
+      // nodeColors[nodeId] = allNodes[nodeId].color;
+      allNodes[nodeId].color = "rgba(200,200,200,0.5)";
+      if (allNodes[nodeId].hiddenLabel === undefined) {
+        allNodes[nodeId].hiddenLabel = allNodes[nodeId].label;
+        allNodes[nodeId].label = undefined;
+      }
+    }
+    var connectedNodes = network.getConnectedNodes(selectedNode);
+    var allConnectedNodes = [];
+
+    // get the second degree nodes
+    for (i = 1; i < degrees; i++) {
+      for (j = 0; j < connectedNodes.length; j++) {
+        allConnectedNodes = allConnectedNodes.concat(
+          network.getConnectedNodes(connectedNodes[j])
+        );
+      }
+    }
+
+    // all second degree nodes get a different color and their label back
+    for (i = 0; i < allConnectedNodes.length; i++) {
+      // allNodes[allConnectedNodes[i]].color = "pink";
+      allNodes[allConnectedNodes[i]].color = "rgba(150,150,150,0.75)";
+      if (allNodes[allConnectedNodes[i]].hiddenLabel !== undefined) {
+        allNodes[allConnectedNodes[i]].label =
+          allNodes[allConnectedNodes[i]].hiddenLabel;
+        allNodes[allConnectedNodes[i]].hiddenLabel = undefined;
+      }
+    }
+
+    // all first degree nodes get their own color and their label back
+    for (i = 0; i < connectedNodes.length; i++) {
+      // allNodes[connectedNodes[i]].color = undefined;
+      allNodes[connectedNodes[i]].color = nodeColors[connectedNodes[i]];
+      if (allNodes[connectedNodes[i]].hiddenLabel !== undefined) {
+        allNodes[connectedNodes[i]].label =
+          allNodes[connectedNodes[i]].hiddenLabel;
+        allNodes[connectedNodes[i]].hiddenLabel = undefined;
+      }
+    }
+
+    // the main node gets its own color and its label back.
+    // allNodes[selectedNode].color = undefined;
+    allNodes[selectedNode].color = nodeColors[selectedNode];
+    if (allNodes[selectedNode].hiddenLabel !== undefined) {
+      allNodes[selectedNode].label = allNodes[selectedNode].hiddenLabel;
+      allNodes[selectedNode].hiddenLabel = undefined;
+    }
+  } else if (highlightActive === true) {
+    // console.log("highlightActive was true");
+    // reset all nodes
+    for (let nodeId in allNodes) {
+      // allNodes[nodeId].color = "purple";
+      allNodes[nodeId].color = nodeColors[nodeId];
+      // delete allNodes[nodeId].color;
+      if (allNodes[nodeId].hiddenLabel !== undefined) {
+        allNodes[nodeId].label = allNodes[nodeId].hiddenLabel;
+        allNodes[nodeId].hiddenLabel = undefined;
+      }
+    }
+    highlightActive = false;
+  }
+
+  // transform the object into an array
+  var updateArray = [];
+  if (params.nodes.length > 0) {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        // console.log(allNodes[nodeId]);
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  } else {
+    // console.log("Nothing was selected");
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        // console.log(allNodes[nodeId]);
+        // allNodes[nodeId].color = {};
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  }
+}
+
+function filterHighlight(params) {
+  allNodes = nodes.get({ returnType: "Object" });
+  // if something is selected:
+  if (params.nodes.length > 0) {
+    filterActive = true;
+    let selectedNodes = params.nodes;
+
+    // hiding all nodes and saving the label
+    for (let nodeId in allNodes) {
+      allNodes[nodeId].hidden = true;
+      if (allNodes[nodeId].savedLabel === undefined) {
+        allNodes[nodeId].savedLabel = allNodes[nodeId].label;
+        allNodes[nodeId].label = undefined;
+      }
+    }
+
+    for (let i=0; i < selectedNodes.length; i++) {
+      allNodes[selectedNodes[i]].hidden = false;
+      if (allNodes[selectedNodes[i]].savedLabel !== undefined) {
+        allNodes[selectedNodes[i]].label = allNodes[selectedNodes[i]].savedLabel;
+        allNodes[selectedNodes[i]].savedLabel = undefined;
+      }
+    }
+
+  } else if (filterActive === true) {
+    // reset all nodes
+    for (let nodeId in allNodes) {
+      allNodes[nodeId].hidden = false;
+      if (allNodes[nodeId].savedLabel !== undefined) {
+        allNodes[nodeId].label = allNodes[nodeId].savedLabel;
+        allNodes[nodeId].savedLabel = undefined;
+      }
+    }
+    filterActive = false;
+  }
+
+  // transform the object into an array
+  var updateArray = [];
+  if (params.nodes.length > 0) {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  } else {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  }
+}
+
+function selectNode(nodes) {
+  network.selectNodes(nodes);
+  neighbourhoodHighlight({ nodes: nodes });
+  return nodes;
+}
+
+function selectNodes(nodes) {
+  network.selectNodes(nodes);
+  filterHighlight({nodes: nodes});
+  return nodes;
+}
+
+function highlightFilter(filter) {
+  let selectedNodes = []
+  let selectedProp = filter['property']
+  if (filter['item'] === 'node') {
+    let allNodes = nodes.get({ returnType: "Object" });
+    for (let nodeId in allNodes) {
+      if (allNodes[nodeId][selectedProp] && filter['value'].includes((allNodes[nodeId][selectedProp]).toString())) {
+        selectedNodes.push(nodeId)
+      }
+    }
+  }
+  else if (filter['item'] === 'edge'){
+    let allEdges = edges.get({returnType: 'object'});
+    // check if the selected property exists for selected edge and select the nodes connected to the edge
+    for (let edge in allEdges) {
+      if (allEdges[edge][selectedProp] && filter['value'].includes((allEdges[edge][selectedProp]).toString())) {
+        selectedNodes.push(allEdges[edge]['from'])
+        selectedNodes.push(allEdges[edge]['to'])
+      }
+    }
+  }
+  selectNodes(selectedNodes)
+}</script>
+            <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/vis-network/9.1.2/dist/dist/vis-network.min.css" integrity="sha512-WgxfT5LWjfszlPHXRmBWHkV2eceiWTOBvrKCNbdgDYTHrT2AeLCGbF4sZlZw3UMN3WtL0tGUoIAKsu8mllg/XA==" crossorigin="anonymous" referrerpolicy="no-referrer" />
+            <script src="https://cdnjs.cloudflare.com/ajax/libs/vis-network/9.1.2/dist/vis-network.min.js" integrity="sha512-LnvoEWDFrqGHlHmDD2101OrLcbsfkrzoSpvtSQtxK3RMnRV0eOkhhBN2dXHKRrUU8p2DGRTk35n4O8nWSVe1mQ==" crossorigin="anonymous" referrerpolicy="no-referrer"></script>
+            
+            
+            
+            
+            
+            
+
+        
+<center>
+<h1></h1>
+</center>
+
+<!-- <link rel="stylesheet" href="../node_modules/vis/dist/vis.min.css" type="text/css" />
+<script type="text/javascript" src="../node_modules/vis/dist/vis.js"> </script>-->
+        <link
+          href="https://cdn.jsdelivr.net/npm/bootstrap@5.0.0-beta3/dist/css/bootstrap.min.css"
+          rel="stylesheet"
+          integrity="sha384-eOJMYsd53ii+scO/bJGFsiCZc+5NDVN2yr8+0RDqr0Ql0h+rP48ckxlpbzKgwra6"
+          crossorigin="anonymous"
+        />
+        <script
+          src="https://cdn.jsdelivr.net/npm/bootstrap@5.0.0-beta3/dist/js/bootstrap.bundle.min.js"
+          integrity="sha384-JEW9xMcG8R+pH31jmWH6WWP0WintQrMb4s7ZOdauHnUtxwoG2vI5DkLtS3qm9Ekf"
+          crossorigin="anonymous"
+        ></script>
+
+
+        <center>
+          <h1></h1>
+        </center>
+        <style type="text/css">
+
+             #mynetwork {
+                 width: 100%;
+                 height: 600px;
+                 background-color: #ffffff;
+                 border: 1px solid lightgray;
+                 position: relative;
+                 float: left;
+             }
+
+             
+
+             
+             #config {
+                 float: left;
+                 width: 400px;
+                 height: 600px;
+             }
+             
+
+             
+        </style>
+    </head>
+
+
+    <body>
+        <div class="card" style="width: 100%">
+            
+            
+            <div id="mynetwork" class="card-body"></div>
+        </div>
+
+        
+        
+            <div id="config"></div>
+        
+
+        <script type="text/javascript">
+
+              // initialize global variables.
+              var edges;
+              var nodes;
+              var allNodes;
+              var allEdges;
+              var nodeColors;
+              var originalNodes;
+              var network;
+              var container;
+              var options, data;
+              var filter = {
+                  item : '',
+                  property : '',
+                  value : []
+              };
+
+              
+
+              
+
+              // This method is responsible for drawing the graph, returns the drawn network
+              function drawGraph() {
+                  var container = document.getElementById('mynetwork');
+
+                  
+
+                  // parsing and collecting nodes and edges from the python
+                  nodes = new vis.DataSet([]);
+                  edges = new vis.DataSet([]);
+
+                  nodeColors = {};
+                  allNodes = nodes.get({ returnType: "Object" });
+                  for (nodeId in allNodes) {
+                    nodeColors[nodeId] = allNodes[nodeId].color;
+                  }
+                  allEdges = edges.get({ returnType: "Object" });
+                  // adding nodes and edges to the graph
+                  data = {nodes: nodes, edges: edges};
+
+                  var options = {
+    "configure": {
+        "enabled": true
+    },
+    "edges": {
+        "color": {
+            "inherit": true
+        },
+        "smooth": {
+            "enabled": true,
+            "type": "dynamic"
+        }
+    },
+    "interaction": {
+        "dragNodes": true,
+        "hideEdgesOnDrag": false,
+        "hideNodesOnDrag": false
+    },
+    "physics": {
+        "enabled": true,
+        "stabilization": {
+            "enabled": true,
+            "fit": true,
+            "iterations": 1000,
+            "onlyDynamicEdges": false,
+            "updateInterval": 50
+        }
+    }
+};
+
+                  
+
+
+                  
+                  // if this network requires displaying the configure window,
+                  // put it in its div
+                  options.configure["container"] = document.getElementById("config");
+                  
+
+                  network = new vis.Network(container, data, options);
+
+                  
+
+                  
+
+                  
+
+
+                  
+
+                  return network;
+
+              }
+              drawGraph();
+        </script>
+    </body>
+</html>
\ No newline at end of file
diff --git a/notebooks/monty.html b/notebooks/monty.html
new file mode 100644
index 0000000..e1628bb
--- /dev/null
+++ b/notebooks/monty.html
@@ -0,0 +1,321 @@
+<html>
+    <head>
+        <meta charset="utf-8">
+        
+            <script>function neighbourhoodHighlight(params) {
+  // console.log("in nieghbourhoodhighlight");
+  allNodes = nodes.get({ returnType: "Object" });
+  // originalNodes = JSON.parse(JSON.stringify(allNodes));
+  // if something is selected:
+  if (params.nodes.length > 0) {
+    highlightActive = true;
+    var i, j;
+    var selectedNode = params.nodes[0];
+    var degrees = 2;
+
+    // mark all nodes as hard to read.
+    for (let nodeId in allNodes) {
+      // nodeColors[nodeId] = allNodes[nodeId].color;
+      allNodes[nodeId].color = "rgba(200,200,200,0.5)";
+      if (allNodes[nodeId].hiddenLabel === undefined) {
+        allNodes[nodeId].hiddenLabel = allNodes[nodeId].label;
+        allNodes[nodeId].label = undefined;
+      }
+    }
+    var connectedNodes = network.getConnectedNodes(selectedNode);
+    var allConnectedNodes = [];
+
+    // get the second degree nodes
+    for (i = 1; i < degrees; i++) {
+      for (j = 0; j < connectedNodes.length; j++) {
+        allConnectedNodes = allConnectedNodes.concat(
+          network.getConnectedNodes(connectedNodes[j])
+        );
+      }
+    }
+
+    // all second degree nodes get a different color and their label back
+    for (i = 0; i < allConnectedNodes.length; i++) {
+      // allNodes[allConnectedNodes[i]].color = "pink";
+      allNodes[allConnectedNodes[i]].color = "rgba(150,150,150,0.75)";
+      if (allNodes[allConnectedNodes[i]].hiddenLabel !== undefined) {
+        allNodes[allConnectedNodes[i]].label =
+          allNodes[allConnectedNodes[i]].hiddenLabel;
+        allNodes[allConnectedNodes[i]].hiddenLabel = undefined;
+      }
+    }
+
+    // all first degree nodes get their own color and their label back
+    for (i = 0; i < connectedNodes.length; i++) {
+      // allNodes[connectedNodes[i]].color = undefined;
+      allNodes[connectedNodes[i]].color = nodeColors[connectedNodes[i]];
+      if (allNodes[connectedNodes[i]].hiddenLabel !== undefined) {
+        allNodes[connectedNodes[i]].label =
+          allNodes[connectedNodes[i]].hiddenLabel;
+        allNodes[connectedNodes[i]].hiddenLabel = undefined;
+      }
+    }
+
+    // the main node gets its own color and its label back.
+    // allNodes[selectedNode].color = undefined;
+    allNodes[selectedNode].color = nodeColors[selectedNode];
+    if (allNodes[selectedNode].hiddenLabel !== undefined) {
+      allNodes[selectedNode].label = allNodes[selectedNode].hiddenLabel;
+      allNodes[selectedNode].hiddenLabel = undefined;
+    }
+  } else if (highlightActive === true) {
+    // console.log("highlightActive was true");
+    // reset all nodes
+    for (let nodeId in allNodes) {
+      // allNodes[nodeId].color = "purple";
+      allNodes[nodeId].color = nodeColors[nodeId];
+      // delete allNodes[nodeId].color;
+      if (allNodes[nodeId].hiddenLabel !== undefined) {
+        allNodes[nodeId].label = allNodes[nodeId].hiddenLabel;
+        allNodes[nodeId].hiddenLabel = undefined;
+      }
+    }
+    highlightActive = false;
+  }
+
+  // transform the object into an array
+  var updateArray = [];
+  if (params.nodes.length > 0) {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        // console.log(allNodes[nodeId]);
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  } else {
+    // console.log("Nothing was selected");
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        // console.log(allNodes[nodeId]);
+        // allNodes[nodeId].color = {};
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  }
+}
+
+function filterHighlight(params) {
+  allNodes = nodes.get({ returnType: "Object" });
+  // if something is selected:
+  if (params.nodes.length > 0) {
+    filterActive = true;
+    let selectedNodes = params.nodes;
+
+    // hiding all nodes and saving the label
+    for (let nodeId in allNodes) {
+      allNodes[nodeId].hidden = true;
+      if (allNodes[nodeId].savedLabel === undefined) {
+        allNodes[nodeId].savedLabel = allNodes[nodeId].label;
+        allNodes[nodeId].label = undefined;
+      }
+    }
+
+    for (let i=0; i < selectedNodes.length; i++) {
+      allNodes[selectedNodes[i]].hidden = false;
+      if (allNodes[selectedNodes[i]].savedLabel !== undefined) {
+        allNodes[selectedNodes[i]].label = allNodes[selectedNodes[i]].savedLabel;
+        allNodes[selectedNodes[i]].savedLabel = undefined;
+      }
+    }
+
+  } else if (filterActive === true) {
+    // reset all nodes
+    for (let nodeId in allNodes) {
+      allNodes[nodeId].hidden = false;
+      if (allNodes[nodeId].savedLabel !== undefined) {
+        allNodes[nodeId].label = allNodes[nodeId].savedLabel;
+        allNodes[nodeId].savedLabel = undefined;
+      }
+    }
+    filterActive = false;
+  }
+
+  // transform the object into an array
+  var updateArray = [];
+  if (params.nodes.length > 0) {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  } else {
+    for (let nodeId in allNodes) {
+      if (allNodes.hasOwnProperty(nodeId)) {
+        updateArray.push(allNodes[nodeId]);
+      }
+    }
+    nodes.update(updateArray);
+  }
+}
+
+function selectNode(nodes) {
+  network.selectNodes(nodes);
+  neighbourhoodHighlight({ nodes: nodes });
+  return nodes;
+}
+
+function selectNodes(nodes) {
+  network.selectNodes(nodes);
+  filterHighlight({nodes: nodes});
+  return nodes;
+}
+
+function highlightFilter(filter) {
+  let selectedNodes = []
+  let selectedProp = filter['property']
+  if (filter['item'] === 'node') {
+    let allNodes = nodes.get({ returnType: "Object" });
+    for (let nodeId in allNodes) {
+      if (allNodes[nodeId][selectedProp] && filter['value'].includes((allNodes[nodeId][selectedProp]).toString())) {
+        selectedNodes.push(nodeId)
+      }
+    }
+  }
+  else if (filter['item'] === 'edge'){
+    let allEdges = edges.get({returnType: 'object'});
+    // check if the selected property exists for selected edge and select the nodes connected to the edge
+    for (let edge in allEdges) {
+      if (allEdges[edge][selectedProp] && filter['value'].includes((allEdges[edge][selectedProp]).toString())) {
+        selectedNodes.push(allEdges[edge]['from'])
+        selectedNodes.push(allEdges[edge]['to'])
+      }
+    }
+  }
+  selectNodes(selectedNodes)
+}</script>
+            <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/vis-network/9.1.2/dist/dist/vis-network.min.css" integrity="sha512-WgxfT5LWjfszlPHXRmBWHkV2eceiWTOBvrKCNbdgDYTHrT2AeLCGbF4sZlZw3UMN3WtL0tGUoIAKsu8mllg/XA==" crossorigin="anonymous" referrerpolicy="no-referrer" />
+            <script src="https://cdnjs.cloudflare.com/ajax/libs/vis-network/9.1.2/dist/vis-network.min.js" integrity="sha512-LnvoEWDFrqGHlHmDD2101OrLcbsfkrzoSpvtSQtxK3RMnRV0eOkhhBN2dXHKRrUU8p2DGRTk35n4O8nWSVe1mQ==" crossorigin="anonymous" referrerpolicy="no-referrer"></script>
+            
+            
+            
+            
+            
+            
+
+        
+<center>
+<h1></h1>
+</center>
+
+<!-- <link rel="stylesheet" href="../node_modules/vis/dist/vis.min.css" type="text/css" />
+<script type="text/javascript" src="../node_modules/vis/dist/vis.js"> </script>-->
+        <link
+          href="https://cdn.jsdelivr.net/npm/bootstrap@5.0.0-beta3/dist/css/bootstrap.min.css"
+          rel="stylesheet"
+          integrity="sha384-eOJMYsd53ii+scO/bJGFsiCZc+5NDVN2yr8+0RDqr0Ql0h+rP48ckxlpbzKgwra6"
+          crossorigin="anonymous"
+        />
+        <script
+          src="https://cdn.jsdelivr.net/npm/bootstrap@5.0.0-beta3/dist/js/bootstrap.bundle.min.js"
+          integrity="sha384-JEW9xMcG8R+pH31jmWH6WWP0WintQrMb4s7ZOdauHnUtxwoG2vI5DkLtS3qm9Ekf"
+          crossorigin="anonymous"
+        ></script>
+
+
+        <center>
+          <h1></h1>
+        </center>
+        <style type="text/css">
+
+             #mynetwork {
+                 width: 100%;
+                 height: 600px;
+                 background-color: #ffffff;
+                 border: 1px solid lightgray;
+                 position: relative;
+                 float: left;
+             }
+
+             
+
+             
+
+             
+        </style>
+    </head>
+
+
+    <body>
+        <div class="card" style="width: 100%">
+            
+            
+            <div id="mynetwork" class="card-body"></div>
+        </div>
+
+        
+        
+
+        <script type="text/javascript">
+
+              // initialize global variables.
+              var edges;
+              var nodes;
+              var allNodes;
+              var allEdges;
+              var nodeColors;
+              var originalNodes;
+              var network;
+              var container;
+              var options, data;
+              var filter = {
+                  item : '',
+                  property : '',
+                  value : []
+              };
+
+              
+
+              
+
+              // This method is responsible for drawing the graph, returns the drawn network
+              function drawGraph() {
+                  var container = document.getElementById('mynetwork');
+
+                  
+
+                  // parsing and collecting nodes and edges from the python
+                  nodes = new vis.DataSet([{"borderWidth": 3, "color": null, "id": 0, "label": "init", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 1, "label": "carchosen", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 2, "label": "carchosen", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 3, "label": "carchosen", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 4, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 5, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 6, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 7, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 8, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 9, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 10, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 11, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 12, "label": "open", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 13, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 14, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 15, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 16, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 17, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 18, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 19, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 20, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 21, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 22, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 23, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 24, "label": "goatrevealed", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 25, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 26, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 27, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 28, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 29, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 30, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 31, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 32, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 33, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 34, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 35, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 36, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 37, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 38, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 39, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 40, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 41, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 42, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 43, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 44, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 45, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 46, "label": "done", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 47, "label": "done,target", "shape": "dot"}, {"borderWidth": 1, "color": null, "id": 48, "label": "done", "shape": "dot"}]);
+                  edges = new vis.DataSet([{"arrows": "to", "color": "red", "from": 0, "label": "1/3", "to": 1}, {"arrows": "to", "color": "red", "from": 0, "label": "1/3", "to": 2}, {"arrows": "to", "color": "red", "from": 0, "label": "1/3", "to": 3}, {"arrows": "to", "color": "red", "from": 4, "label": "1/2", "to": 13}, {"arrows": "to", "color": "red", "from": 4, "label": "1/2", "to": 14}, {"arrows": "to", "color": "red", "from": 5, "label": "1", "to": 15}, {"arrows": "to", "color": "red", "from": 6, "label": "1", "to": 16}, {"arrows": "to", "color": "red", "from": 7, "label": "1", "to": 17}, {"arrows": "to", "color": "red", "from": 8, "label": "1/2", "to": 18}, {"arrows": "to", "color": "red", "from": 8, "label": "1/2", "to": 19}, {"arrows": "to", "color": "red", "from": 9, "label": "1", "to": 20}, {"arrows": "to", "color": "red", "from": 10, "label": "1", "to": 21}, {"arrows": "to", "color": "red", "from": 11, "label": "1", "to": 22}, {"arrows": "to", "color": "red", "from": 12, "label": "1/2", "to": 23}, {"arrows": "to", "color": "red", "from": 12, "label": "1/2", "to": 24}, {"arrows": "to", "color": "red", "from": 25, "label": "1", "to": 25}, {"arrows": "to", "color": "red", "from": 26, "label": "1", "to": 26}, {"arrows": "to", "color": "red", "from": 27, "label": "1", "to": 27}, {"arrows": "to", "color": "red", "from": 28, "label": "1", "to": 28}, {"arrows": "to", "color": "red", "from": 29, "label": "1", "to": 29}, {"arrows": "to", "color": "red", "from": 30, "label": "1", "to": 30}, {"arrows": "to", "color": "red", "from": 31, "label": "1", "to": 31}, {"arrows": "to", "color": "red", "from": 32, "label": "1", "to": 32}, {"arrows": "to", "color": "red", "from": 33, "label": "1", "to": 33}, {"arrows": "to", "color": "red", "from": 34, "label": "1", "to": 34}, {"arrows": "to", "color": "red", "from": 35, "label": "1", "to": 35}, {"arrows": "to", "color": "red", "from": 36, "label": "1", "to": 36}, {"arrows": "to", "color": "red", "from": 37, "label": "1", "to": 37}, {"arrows": "to", "color": "red", "from": 38, "label": "1", "to": 38}, {"arrows": "to", "color": "red", "from": 39, "label": "1", "to": 39}, {"arrows": "to", "color": "red", "from": 40, "label": "1", "to": 40}, {"arrows": "to", "color": "red", "from": 41, "label": "1", "to": 41}, {"arrows": "to", "color": "red", "from": 42, "label": "1", "to": 42}, {"arrows": "to", "color": "red", "from": 43, "label": "1", "to": 43}, {"arrows": "to", "color": "red", "from": 44, "label": "1", "to": 44}, {"arrows": "to", "color": "red", "from": 45, "label": "1", "to": 45}, {"arrows": "to", "color": "red", "from": 46, "label": "1", "to": 46}, {"arrows": "to", "color": "red", "from": 47, "label": "1", "to": 47}, {"arrows": "to", "color": "red", "from": 48, "label": "1", "to": 48}]);
+
+                  nodeColors = {};
+                  allNodes = nodes.get({ returnType: "Object" });
+                  for (nodeId in allNodes) {
+                    nodeColors[nodeId] = allNodes[nodeId].color;
+                  }
+                  allEdges = edges.get({ returnType: "Object" });
+                  // adding nodes and edges to the graph
+                  data = {nodes: nodes, edges: edges};
+
+                  var options = {"nodes": {"color": {"background": "white", "border": "black"}}, "edge": {"color": "blue"}};
+
+                  
+
+
+                  
+
+                  network = new vis.Network(container, data, options);
+
+                  
+
+                  
+
+                  
+
+
+                  
+
+                  return network;
+
+              }
+              drawGraph();
+        </script>
+    </body>
+</html>
\ No newline at end of file
diff --git a/notebooks/monty_hall.ipynb b/notebooks/monty_hall.ipynb
new file mode 100644
index 0000000..4be0cc9
--- /dev/null
+++ b/notebooks/monty_hall.ipynb
@@ -0,0 +1,131 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "bce64a65-ea2a-42b6-adac-0533b7541ca0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import stormvogel.model\n",
+    "\n",
+    "mdp = stormvogel.model.new_mdp(\"Monty Hall\")\n",
+    "\n",
+    "init = mdp.get_initial_state()\n",
+    "\n",
+    "# first choose car position\n",
+    "init.set_transitions(\n",
+    "    [(1 / 3, mdp.new_state(\"carchosen\", {\"car_pos\": i})) for i in range(3)]\n",
+    ")\n",
+    "\n",
+    "# we choose a door in each case\n",
+    "for s in mdp.get_states_with(\"carchosen\"):\n",
+    "    s.set_transitions(\n",
+    "        [\n",
+    "            (\n",
+    "                mdp.action(f\"open{i}\"),\n",
+    "                mdp.new_state(\"open\", s.features | {\"chosen_pos\": i}),\n",
+    "            )\n",
+    "            for i in range(3)\n",
+    "        ]\n",
+    "    )\n",
+    "\n",
+    "# the other goat is revealed\n",
+    "for s in mdp.get_states_with(\"open\"):\n",
+    "    car_pos = s.features[\"car_pos\"]\n",
+    "    chosen_pos = s.features[\"chosen_pos\"]\n",
+    "    other_pos = {0, 1, 2} - {car_pos, chosen_pos}\n",
+    "    s.set_transitions(\n",
+    "        [\n",
+    "            (\n",
+    "                1 / len(other_pos),\n",
+    "                mdp.new_state(\"goatrevealed\", s.features | {\"reveal_pos\": i}),\n",
+    "            )\n",
+    "            for i in other_pos\n",
+    "        ]\n",
+    "    )\n",
+    "\n",
+    "# we must choose whether we want to switch\n",
+    "for s in mdp.get_states_with(\"goatrevealed\"):\n",
+    "    car_pos = s.features[\"car_pos\"]\n",
+    "    chosen_pos = s.features[\"chosen_pos\"]\n",
+    "    reveal_pos = s.features[\"reveal_pos\"]\n",
+    "    other_pos = list({0, 1, 2} - {reveal_pos, chosen_pos})[0]\n",
+    "    s.set_transitions(\n",
+    "        [\n",
+    "            (\n",
+    "                mdp.action(\"stay\"),\n",
+    "                mdp.new_state(\n",
+    "                    [\"done\"] + ([\"target\"] if chosen_pos == car_pos else []),\n",
+    "                    s.features | {\"chosen_pos\": chosen_pos},\n",
+    "                ),\n",
+    "            ),\n",
+    "            (\n",
+    "                mdp.action(\"switch\"),\n",
+    "                mdp.new_state(\n",
+    "                    [\"done\"] + ([\"target\"] if other_pos == car_pos else []),\n",
+    "                    s.features | {\"chosen_pos\": other_pos},\n",
+    "                ),\n",
+    "            ),\n",
+    "        ]\n",
+    "    )\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "id": "02b68164-00de-41b1-ba2b-65e12bef9032",
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "NotImplementedError",
+     "evalue": "Non-empty actions are not supported yet.",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mNotImplementedError\u001b[0m                       Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[4], line 3\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mstormvogel\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mvisualization\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m show\n\u001b[0;32m----> 3\u001b[0m \u001b[43mshow\u001b[49m\u001b[43m(\u001b[49m\u001b[43mmdp\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mmonty\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m~/git/stormvogel/.venv/lib/python3.11/site-packages/stormvogel/visualization.py:75\u001b[0m, in \u001b[0;36mshow\u001b[0;34m(model, name, notebook)\u001b[0m\n\u001b[1;32m     67\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mshow\u001b[39m(model: Model, name: \u001b[38;5;28mstr\u001b[39m, notebook: \u001b[38;5;28mbool\u001b[39m\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mTrue\u001b[39;00m):\n\u001b[1;32m     68\u001b[0m \u001b[38;5;250m    \u001b[39m\u001b[38;5;124;03m\"\"\"Create visualization of a Model using a pyvis Network\u001b[39;00m\n\u001b[1;32m     69\u001b[0m \n\u001b[1;32m     70\u001b[0m \u001b[38;5;124;03m    Args:\u001b[39;00m\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m     73\u001b[0m \u001b[38;5;124;03m        notebook (bool, optional): Leave to true if you are using in a notebook. Defaults to True.\u001b[39;00m\n\u001b[1;32m     74\u001b[0m \u001b[38;5;124;03m    \"\"\"\u001b[39;00m\n\u001b[0;32m---> 75\u001b[0m     vis \u001b[38;5;241m=\u001b[39m \u001b[43mVisualization\u001b[49m\u001b[43m(\u001b[49m\u001b[43mmodel\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mname\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mnotebook\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m     76\u001b[0m     vis\u001b[38;5;241m.\u001b[39mshow()\n",
+      "File \u001b[0;32m~/git/stormvogel/.venv/lib/python3.11/site-packages/stormvogel/visualization.py:24\u001b[0m, in \u001b[0;36mVisualization.__init__\u001b[0;34m(self, model, name, notebook, cdn_resources)\u001b[0m\n\u001b[1;32m     22\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mg \u001b[38;5;241m=\u001b[39m Network(notebook\u001b[38;5;241m=\u001b[39mnotebook, directed\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mTrue\u001b[39;00m, cdn_resources\u001b[38;5;241m=\u001b[39mcdn_resources)\n\u001b[1;32m     23\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m__add_nodes()\n\u001b[0;32m---> 24\u001b[0m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m__add_transitions\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m     25\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m__set_layout()\n",
+      "File \u001b[0;32m~/git/stormvogel/.venv/lib/python3.11/site-packages/stormvogel/visualization.py:62\u001b[0m, in \u001b[0;36mVisualization.__add_transitions\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m     60\u001b[0m         \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mg\u001b[38;5;241m.\u001b[39madd_edge(state_id, target\u001b[38;5;241m.\u001b[39mid, color\u001b[38;5;241m=\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mred\u001b[39m\u001b[38;5;124m\"\u001b[39m, label\u001b[38;5;241m=\u001b[39m\u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39m__formatted_prob(prob))\n\u001b[1;32m     61\u001b[0m \u001b[38;5;28;01melse\u001b[39;00m:\n\u001b[0;32m---> 62\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m \u001b[38;5;167;01mNotImplementedError\u001b[39;00m(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mNon-empty actions are not supported yet.\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n",
+      "\u001b[0;31mNotImplementedError\u001b[0m: Non-empty actions are not supported yet."
+     ]
+    }
+   ],
+   "source": [
+    "from stormvogel.visualization import show\n",
+    "\n",
+    "show(mdp, \"monty\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3ec68bb5-2728-458e-8d11-6015f80d79f7",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}
diff --git a/stormvogel/visualization.py b/stormvogel/visualization.py
index 26fb39a..cd7d39d 100644
--- a/stormvogel/visualization.py
+++ b/stormvogel/visualization.py
@@ -1,12 +1,80 @@
 """Contains stuff for visualization"""
-
-from stormvogel.model import Model
+from pyvis.network import Network
+from stormvogel.model import Model, EmptyAction
 from ipywidgets import interact
+from IPython.display import display
+from fractions import Fraction
+
+class Visualization:
+    """Handles visualization of a Model using a pyvis Network."""
+    def __init__(self, model: Model, name: str, notebook: bool=True, cdn_resources: str="remote") -> None:
+        """Create visualization of a Model using a pyvis Network
+
+        Args:
+            model (Model): The stormvogel model to be displayed.
+            name (str): The name of the resulting html file.
+            notebook (bool, optional): Leave to true if you are using in a notebook. Defaults to True.
+        """
+        self.model = model
+        if name[-5:] != ".html": # We do not require the user to explicitly type .html in their names.
+            name += ".html"
+        self.name = name
+        self.g = Network(notebook=notebook, directed=True, cdn_resources=cdn_resources)
+        self.__add_nodes()
+        self.__add_transitions()
+        self.__set_layout()
+    
+    def __set_layout(self):
+        self.g.set_options("""
+var options = {
+    "nodes": {
+        "color": {
+            "background": "white",
+            "border": "black"
+        }
+    },
+    "edge": {
+        "color": "blue"
+    }
+}""")
+    
+    def __add_nodes(self):
+        """For each state in the model, add a node to the graph."""
+        for state in self.model.states.values():
+            borderWidth = 1
+            if state == self.model.get_initial_state():
+                borderWidth = 3
+            self.g.add_node(state.id, label=",".join(state.labels), color=None, borderWidth=borderWidth)
+    
+    def __formatted_prob(self, prob: float) -> str:
+        """Take a probability value and format it nicely"""
+        return str(Fraction(prob).limit_denominator(20))
+    
+    def __add_transitions(self):
+        """For each transition in the model, add a transition in the graph."""
+        for state_id, transition in self.model.transitions.items():
+            for action, branch in transition.transition.items():
+                if action == EmptyAction:
+                    # Only draw probabilities
+                    for prob, target in branch.branch:
+                        self.g.add_edge(state_id, target.id, color="red", label=self.__formatted_prob(prob))
+                else:
+                    raise NotImplementedError("Non-empty actions are not supported yet.")
 
+    def show(self):
+        """Show the constructed model"""
+        display(self.g.show(name=self.name))
 
-def show(m: Model):
-    pass
+def show(model: Model, name: str, notebook: bool=True):
+    """Create visualization of a Model using a pyvis Network
 
+    Args:
+        model (Model): The stormvogel model to be displayed.
+        name (str): The name of the resulting html file.
+        notebook (bool, optional): Leave to true if you are using in a notebook. Defaults to True.
+    """
+    vis = Visualization(model, name, notebook)
+    vis.show()
 
 def make_slider():
     return interact(lambda x: x, x=10)