diff --git a/images/sample-1.svg b/images/sample-1.svg
index d32b046..1434367 100644
--- a/images/sample-1.svg
+++ b/images/sample-1.svg
@@ -9,9 +9,9 @@
 <g id="graph0" class="graph" transform="scale(1 1) rotate(0) translate(4 70)">
 <title>G</title>
 <polygon fill="white" stroke="transparent" points="-4,4 -4,-70 144,-70 144,4 -4,4"/>
-<!-- leaf140273375911024 -->
+<!-- leaf140728778297408 -->
 <g id="node1" class="node">
-<title>leaf140273375911024</title>
+<title>leaf140728778297408</title>
 <polygon fill="none" stroke="black" stroke-width="0" points="15,-66 0,-66 0,0 15,0 15,-66"/>
 <text text-anchor="start" x="-16" y="-55.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 <text text-anchor="start" x="0" y="-55.8" font-family="Consolas" font-size="9.00" fill="#444443">2x1</text>
@@ -22,44 +22,44 @@
 <text text-anchor="start" x="4.5" y="-30.3" font-family="Consolas" font-size="12.00" fill="#444443">a</text>
 <text text-anchor="start" x="11.5" y="-30.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 </g>
-<!-- leaf140273375909152 -->
+<!-- leaf140728778297504 -->
 <g id="node2" class="node">
-<title>leaf140273375909152</title>
+<title>leaf140728778297504</title>
 <polygon fill="none" stroke="black" stroke-width="0" points="34.5,-51 16.5,-51 16.5,-15 34.5,-15 34.5,-51"/>
 <text text-anchor="start" x="22" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">=</text>
 </g>
-<!-- leaf140273375911024&#45;&gt;leaf140273375909152 -->
-<!-- leaf140273375909584 -->
+<!-- leaf140728778297408&#45;&gt;leaf140728778297504 -->
+<!-- leaf140728778299904 -->
 <g id="node3" class="node">
-<title>leaf140273375909584</title>
+<title>leaf140728778299904</title>
 <polygon fill="none" stroke="black" stroke-width="0" points="69.5,-51 35.5,-51 35.5,-15 69.5,-15 69.5,-51"/>
 <text text-anchor="start" x="35.5" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">torch</text>
 </g>
-<!-- leaf140273375909152&#45;&gt;leaf140273375909584 -->
-<!-- leaf140273375909680 -->
+<!-- leaf140728778297504&#45;&gt;leaf140728778299904 -->
+<!-- leaf140728778300000 -->
 <g id="node4" class="node">
-<title>leaf140273375909680</title>
+<title>leaf140728778300000</title>
 <polygon fill="none" stroke="black" stroke-width="0" points="78,-51 71,-51 71,-15 78,-15 78,-51"/>
 <text text-anchor="start" x="71" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">.</text>
 </g>
-<!-- leaf140273375909584&#45;&gt;leaf140273375909680 -->
-<!-- leaf140273375909776 -->
+<!-- leaf140728778299904&#45;&gt;leaf140728778300000 -->
+<!-- leaf140728778300048 -->
 <g id="node5" class="node">
-<title>leaf140273375909776</title>
+<title>leaf140728778300048</title>
 <polygon fill="none" stroke="black" stroke-width="0" points="107.5,-51 79.5,-51 79.5,-15 107.5,-15 107.5,-51"/>
 <text text-anchor="start" x="79.5" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">relu</text>
 </g>
-<!-- leaf140273375909680&#45;&gt;leaf140273375909776 -->
-<!-- leaf140273375909872 -->
+<!-- leaf140728778300000&#45;&gt;leaf140728778300048 -->
+<!-- leaf140728778300144 -->
 <g id="node6" class="node">
-<title>leaf140273375909872</title>
+<title>leaf140728778300144</title>
 <polygon fill="none" stroke="black" stroke-width="0" points="116,-51 109,-51 109,-15 116,-15 116,-51"/>
 <text text-anchor="start" x="109" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">(</text>
 </g>
-<!-- leaf140273375909776&#45;&gt;leaf140273375909872 -->
-<!-- leaf140273375909104 -->
+<!-- leaf140728778300048&#45;&gt;leaf140728778300144 -->
+<!-- leaf140728778300192 -->
 <g id="node7" class="node">
-<title>leaf140273375909104</title>
+<title>leaf140728778300192</title>
 <polygon fill="none" stroke="black" stroke-width="0" points="132,-66 117,-66 117,0 132,0 132,-66"/>
 <text text-anchor="start" x="101" y="-55.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 <text text-anchor="start" x="117" y="-55.8" font-family="Consolas" font-size="9.00" fill="#444443">2x1</text>
@@ -70,13 +70,13 @@
 <text text-anchor="start" x="121.5" y="-30.3" font-family="Consolas" font-size="12.00" fill="#444443">x</text>
 <text text-anchor="start" x="128.5" y="-30.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 </g>
-<!-- leaf140273375909872&#45;&gt;leaf140273375909104 -->
-<!-- leaf140273375910016 -->
+<!-- leaf140728778300144&#45;&gt;leaf140728778300192 -->
+<!-- leaf140728778300288 -->
 <g id="node8" class="node">
-<title>leaf140273375910016</title>
+<title>leaf140728778300288</title>
 <polygon fill="none" stroke="black" stroke-width="0" points="140,-51 133,-51 133,-15 140,-15 140,-51"/>
 <text text-anchor="start" x="133" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">)</text>
 </g>
-<!-- leaf140273375909104&#45;&gt;leaf140273375910016 -->
+<!-- leaf140728778300192&#45;&gt;leaf140728778300288 -->
 </g>
 </svg>
diff --git a/images/sample-2.svg b/images/sample-2.svg
index ce3ad02..40d98fa 100644
--- a/images/sample-2.svg
+++ b/images/sample-2.svg
@@ -4,128 +4,128 @@
 <!-- Generated by graphviz version 2.44.1 (20200629.0846)
  -->
 <!-- Title: G Pages: 1 -->
-<svg width="275pt" height="94pt"
- viewBox="0.00 0.00 275.00 94.00" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">
-<g id="graph0" class="graph" transform="scale(1 1) rotate(0) translate(4 90)">
+<svg width="275pt" height="74pt"
+ viewBox="0.00 0.00 275.00 74.00" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">
+<g id="graph0" class="graph" transform="scale(1 1) rotate(0) translate(4 70)">
 <title>G</title>
-<polygon fill="white" stroke="transparent" points="-4,4 -4,-90 271,-90 271,4 -4,4"/>
-<!-- leaf140273375910448 -->
+<polygon fill="white" stroke="transparent" points="-4,4 -4,-70 271,-70 271,4 -4,4"/>
+<!-- leaf140728778300816 -->
 <g id="node1" class="node">
-<title>leaf140273375910448</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="15,-86 0,-86 0,0 15,0 15,-86"/>
-<text text-anchor="start" x="-16" y="-75.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="0" y="-75.8" font-family="Consolas" font-size="9.00" fill="#444443">3x1</text>
-<text text-anchor="start" x="16" y="-75.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<polygon fill="#cfe2d4" stroke="transparent" points="0.5,-13 0.5,-73 15.5,-73 15.5,-13 0.5,-13"/>
-<polygon fill="none" stroke="black" points="0.5,-13 0.5,-73 15.5,-73 15.5,-13 0.5,-13"/>
-<text text-anchor="start" x="-11.5" y="-40.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="4.5" y="-40.3" font-family="Consolas" font-size="12.00" fill="#444443">b</text>
-<text text-anchor="start" x="11.5" y="-40.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<title>leaf140728778300816</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="15,-66 0,-66 0,0 15,0 15,-66"/>
+<text text-anchor="start" x="-16" y="-55.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="0" y="-55.8" font-family="Consolas" font-size="9.00" fill="#444443">2x1</text>
+<text text-anchor="start" x="16" y="-55.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<polygon fill="#cfe2d4" stroke="transparent" points="0.5,-13 0.5,-53 15.5,-53 15.5,-13 0.5,-13"/>
+<polygon fill="none" stroke="black" points="0.5,-13 0.5,-53 15.5,-53 15.5,-13 0.5,-13"/>
+<text text-anchor="start" x="-11.5" y="-30.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="4.5" y="-30.3" font-family="Consolas" font-size="12.00" fill="#444443">b</text>
+<text text-anchor="start" x="11.5" y="-30.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 </g>
-<!-- leaf140273375910544 -->
+<!-- leaf140728778299856 -->
 <g id="node2" class="node">
-<title>leaf140273375910544</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="34.5,-61 16.5,-61 16.5,-25 34.5,-25 34.5,-61"/>
-<text text-anchor="start" x="22" y="-39.9" font-family="Consolas" font-size="12.00" fill="#444443">=</text>
+<title>leaf140728778299856</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="34.5,-51 16.5,-51 16.5,-15 34.5,-15 34.5,-51"/>
+<text text-anchor="start" x="22" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">=</text>
 </g>
-<!-- leaf140273375910448&#45;&gt;leaf140273375910544 -->
-<!-- leaf140273375909440 -->
+<!-- leaf140728778300816&#45;&gt;leaf140728778299856 -->
+<!-- leaf140728778299376 -->
 <g id="node3" class="node">
-<title>leaf140273375909440</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="75.5,-86 35.5,-86 35.5,0 75.5,0 75.5,-86"/>
-<text text-anchor="start" x="31.5" y="-75.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="47.5" y="-75.8" font-family="Consolas" font-size="9.00" fill="#444443">3x2</text>
-<text text-anchor="start" x="63.5" y="-75.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<polygon fill="#cfe2d4" stroke="transparent" points="35.5,-13 35.5,-73 75.5,-73 75.5,-13 35.5,-13"/>
-<polygon fill="none" stroke="black" points="35.5,-13 35.5,-73 75.5,-73 75.5,-13 35.5,-13"/>
-<text text-anchor="start" x="36" y="-40.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="52" y="-40.3" font-family="Consolas" font-size="12.00" fill="#444443">W</text>
-<text text-anchor="start" x="59" y="-40.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<title>leaf140728778299376</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="75.5,-66 35.5,-66 35.5,0 75.5,0 75.5,-66"/>
+<text text-anchor="start" x="31.5" y="-55.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="47.5" y="-55.8" font-family="Consolas" font-size="9.00" fill="#444443">2x2</text>
+<text text-anchor="start" x="63.5" y="-55.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<polygon fill="#cfe2d4" stroke="transparent" points="35.5,-13 35.5,-53 75.5,-53 75.5,-13 35.5,-13"/>
+<polygon fill="none" stroke="black" points="35.5,-13 35.5,-53 75.5,-53 75.5,-13 35.5,-13"/>
+<text text-anchor="start" x="36" y="-30.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="52" y="-30.3" font-family="Consolas" font-size="12.00" fill="#444443">W</text>
+<text text-anchor="start" x="59" y="-30.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 </g>
-<!-- leaf140273375910544&#45;&gt;leaf140273375909440 -->
-<!-- leaf140273375909920 -->
+<!-- leaf140728778299856&#45;&gt;leaf140728778299376 -->
+<!-- leaf140728778300384 -->
 <g id="node4" class="node">
-<title>leaf140273375909920</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="90.5,-61 76.5,-61 76.5,-25 90.5,-25 90.5,-61"/>
-<text text-anchor="start" x="80" y="-39.9" font-family="Consolas" font-size="12.00" fill="#444443">@</text>
+<title>leaf140728778300384</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="90.5,-51 76.5,-51 76.5,-15 90.5,-15 90.5,-51"/>
+<text text-anchor="start" x="80" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">@</text>
 </g>
-<!-- leaf140273375909440&#45;&gt;leaf140273375909920 -->
-<!-- leaf140273375910112 -->
+<!-- leaf140728778299376&#45;&gt;leaf140728778300384 -->
+<!-- leaf140728778300960 -->
 <g id="node5" class="node">
-<title>leaf140273375910112</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="107,-76 92,-76 92,-10 107,-10 107,-76"/>
-<text text-anchor="start" x="76" y="-65.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="92" y="-65.8" font-family="Consolas" font-size="9.00" fill="#444443">2x1</text>
-<text text-anchor="start" x="108" y="-65.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<polygon fill="#cfe2d4" stroke="transparent" points="92.5,-23 92.5,-63 107.5,-63 107.5,-23 92.5,-23"/>
-<polygon fill="none" stroke="black" points="92.5,-23 92.5,-63 107.5,-63 107.5,-23 92.5,-23"/>
-<text text-anchor="start" x="80.5" y="-40.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="96.5" y="-40.3" font-family="Consolas" font-size="12.00" fill="#444443">b</text>
-<text text-anchor="start" x="103.5" y="-40.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<title>leaf140728778300960</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="107,-66 92,-66 92,0 107,0 107,-66"/>
+<text text-anchor="start" x="76" y="-55.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="92" y="-55.8" font-family="Consolas" font-size="9.00" fill="#444443">2x1</text>
+<text text-anchor="start" x="108" y="-55.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<polygon fill="#cfe2d4" stroke="transparent" points="92.5,-13 92.5,-53 107.5,-53 107.5,-13 92.5,-13"/>
+<polygon fill="none" stroke="black" points="92.5,-13 92.5,-53 107.5,-53 107.5,-13 92.5,-13"/>
+<text text-anchor="start" x="80.5" y="-30.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="96.5" y="-30.3" font-family="Consolas" font-size="12.00" fill="#444443">b</text>
+<text text-anchor="start" x="103.5" y="-30.3" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 </g>
-<!-- leaf140273375909920&#45;&gt;leaf140273375910112 -->
-<!-- leaf140273375910688 -->
+<!-- leaf140728778300384&#45;&gt;leaf140728778300960 -->
+<!-- leaf140728778300912 -->
 <g id="node6" class="node">
-<title>leaf140273375910688</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="137,-61 108,-61 108,-25 137,-25 137,-61"/>
-<text text-anchor="start" x="119" y="-39.9" font-family="Consolas" font-size="12.00" fill="#444443">+</text>
+<title>leaf140728778300912</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="137,-51 108,-51 108,-15 137,-15 137,-51"/>
+<text text-anchor="start" x="119" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">+</text>
 </g>
-<!-- leaf140273375910112&#45;&gt;leaf140273375910688 -->
-<!-- leaf140273375910736 -->
+<!-- leaf140728778300960&#45;&gt;leaf140728778300912 -->
+<!-- leaf140728778301008 -->
 <g id="node7" class="node">
-<title>leaf140273375910736</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="178.5,-63.5 138.5,-63.5 138.5,-22.5 178.5,-22.5 178.5,-63.5"/>
-<text text-anchor="start" x="139.5" y="-52.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="155.5" y="-52.8" font-family="Consolas" font-size="9.00" fill="#444443">2</text>
-<text text-anchor="start" x="161.5" y="-52.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<polygon fill="#fefecd" stroke="transparent" points="138.5,-35 138.5,-50 178.5,-50 178.5,-35 138.5,-35"/>
-<polygon fill="none" stroke="black" points="138.5,-35 138.5,-50 178.5,-50 178.5,-35 138.5,-35"/>
-<text text-anchor="start" x="139" y="-38.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="155" y="-38.8" font-family="Consolas" font-size="12.00" fill="#444443">h</text>
-<text text-anchor="start" x="162" y="-38.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<title>leaf140728778301008</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="178.5,-53.5 138.5,-53.5 138.5,-12.5 178.5,-12.5 178.5,-53.5"/>
+<text text-anchor="start" x="139.5" y="-42.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="155.5" y="-42.8" font-family="Consolas" font-size="9.00" fill="#444443">2</text>
+<text text-anchor="start" x="161.5" y="-42.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<polygon fill="#fefecd" stroke="transparent" points="138.5,-25 138.5,-40 178.5,-40 178.5,-25 138.5,-25"/>
+<polygon fill="none" stroke="black" points="138.5,-25 138.5,-40 178.5,-40 178.5,-25 138.5,-25"/>
+<text text-anchor="start" x="139" y="-28.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="155" y="-28.8" font-family="Consolas" font-size="12.00" fill="#444443">h</text>
+<text text-anchor="start" x="162" y="-28.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 </g>
-<!-- leaf140273375910688&#45;&gt;leaf140273375910736 -->
-<!-- leaf140273375910832 -->
+<!-- leaf140728778300912&#45;&gt;leaf140728778301008 -->
+<!-- leaf140728778301104 -->
 <g id="node8" class="node">
-<title>leaf140273375910832</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="187,-61 180,-61 180,-25 187,-25 187,-61"/>
-<text text-anchor="start" x="180" y="-39.9" font-family="Consolas" font-size="12.00" fill="#444443">.</text>
+<title>leaf140728778301104</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="187,-51 180,-51 180,-15 187,-15 187,-51"/>
+<text text-anchor="start" x="180" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">.</text>
 </g>
-<!-- leaf140273375910736&#45;&gt;leaf140273375910832 -->
-<!-- leaf140273375910928 -->
+<!-- leaf140728778301008&#45;&gt;leaf140728778301104 -->
+<!-- leaf140728778301296 -->
 <g id="node9" class="node">
-<title>leaf140273375910928</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="209,-61 188,-61 188,-25 209,-25 209,-61"/>
-<text text-anchor="start" x="188" y="-39.9" font-family="Consolas" font-size="12.00" fill="#444443">dot</text>
+<title>leaf140728778301296</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="209,-51 188,-51 188,-15 209,-15 209,-51"/>
+<text text-anchor="start" x="188" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">dot</text>
 </g>
-<!-- leaf140273375910832&#45;&gt;leaf140273375910928 -->
-<!-- leaf140273375911120 -->
+<!-- leaf140728778301104&#45;&gt;leaf140728778301296 -->
+<!-- leaf140728778301392 -->
 <g id="node10" class="node">
-<title>leaf140273375911120</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="217,-61 210,-61 210,-25 217,-25 217,-61"/>
-<text text-anchor="start" x="210" y="-39.9" font-family="Consolas" font-size="12.00" fill="#444443">(</text>
+<title>leaf140728778301392</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="217,-51 210,-51 210,-15 217,-15 217,-51"/>
+<text text-anchor="start" x="210" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">(</text>
 </g>
-<!-- leaf140273375910928&#45;&gt;leaf140273375911120 -->
-<!-- leaf140273375911216 -->
+<!-- leaf140728778301296&#45;&gt;leaf140728778301392 -->
+<!-- leaf140728778298512 -->
 <g id="node11" class="node">
-<title>leaf140273375911216</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="258.5,-63.5 218.5,-63.5 218.5,-22.5 258.5,-22.5 258.5,-63.5"/>
-<text text-anchor="start" x="219.5" y="-52.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="235.5" y="-52.8" font-family="Consolas" font-size="9.00" fill="#444443">2</text>
-<text text-anchor="start" x="241.5" y="-52.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<polygon fill="#fefecd" stroke="transparent" points="218.5,-35 218.5,-50 258.5,-50 258.5,-35 218.5,-35"/>
-<polygon fill="none" stroke="black" points="218.5,-35 218.5,-50 258.5,-50 258.5,-35 218.5,-35"/>
-<text text-anchor="start" x="219" y="-38.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
-<text text-anchor="start" x="235" y="-38.8" font-family="Consolas" font-size="12.00" fill="#444443">h</text>
-<text text-anchor="start" x="242" y="-38.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<title>leaf140728778298512</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="258.5,-53.5 218.5,-53.5 218.5,-12.5 258.5,-12.5 258.5,-53.5"/>
+<text text-anchor="start" x="219.5" y="-42.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="235.5" y="-42.8" font-family="Consolas" font-size="9.00" fill="#444443">2</text>
+<text text-anchor="start" x="241.5" y="-42.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<polygon fill="#fefecd" stroke="transparent" points="218.5,-25 218.5,-40 258.5,-40 258.5,-25 218.5,-25"/>
+<polygon fill="none" stroke="black" points="218.5,-25 218.5,-40 258.5,-40 258.5,-25 218.5,-25"/>
+<text text-anchor="start" x="219" y="-28.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
+<text text-anchor="start" x="235" y="-28.8" font-family="Consolas" font-size="12.00" fill="#444443">h</text>
+<text text-anchor="start" x="242" y="-28.8" font-family="Times,serif" font-size="14.00"> &#160;&#160;&#160;</text>
 </g>
-<!-- leaf140273375911120&#45;&gt;leaf140273375911216 -->
-<!-- leaf140273375911312 -->
+<!-- leaf140728778301392&#45;&gt;leaf140728778298512 -->
+<!-- leaf140728640009264 -->
 <g id="node12" class="node">
-<title>leaf140273375911312</title>
-<polygon fill="none" stroke="black" stroke-width="0" points="267,-61 260,-61 260,-25 267,-25 267,-61"/>
-<text text-anchor="start" x="260" y="-39.9" font-family="Consolas" font-size="12.00" fill="#444443">)</text>
+<title>leaf140728640009264</title>
+<polygon fill="none" stroke="black" stroke-width="0" points="267,-51 260,-51 260,-15 267,-15 267,-51"/>
+<text text-anchor="start" x="260" y="-29.9" font-family="Consolas" font-size="12.00" fill="#444443">)</text>
 </g>
-<!-- leaf140273375911216&#45;&gt;leaf140273375911312 -->
+<!-- leaf140728778298512&#45;&gt;leaf140728640009264 -->
 </g>
 </svg>
diff --git a/testing/viz_testing.py b/testing/viz_testing.py
index 3e78af9..710073a 100644
--- a/testing/viz_testing.py
+++ b/testing/viz_testing.py
@@ -15,10 +15,10 @@ def __init__(self):
         self.a = 3
 
 
-W = torch.tensor([[1, 2], [3, 4], [5, 6]])
+W = torch.tensor([[1, 2], [3, 4]])
 b = torch.tensor([9, 10]).reshape(2, 1)
 x = torch.tensor([4, 5]).reshape(2, 1)
-h = torch.tensor([1, 2])
+h = torch.tensor([1,2])
 a = 3
 
 with tsensor.explain(savefig="/tmp/sample") as e: