diff --git a/abp/fancy.py b/abp/fancy.py
index 7bb276f..c8d1c25 100644
--- a/abp/fancy.py
+++ b/abp/fancy.py
@@ -48,13 +48,13 @@ class GraphState(graphstate.GraphState, networkx.Graph):
             #self.ws.close()
             #self.connect_to_server()
 
-    def layout(self, dim=2):
+    def layout(self):
         """ Automatically lay out the graph """
-        pos = networkx.spring_layout(self, dim, scale=np.sqrt(self.order()))
+        pos = networkx.spring_layout(self, 3, scale=np.sqrt(self.order()))
         middle = np.average(pos.values(), axis=0)
         pos = {key: value - middle for key, value in pos.items()}
-        for key, (x, y) in pos.items():
-            self.node[key]["position"] = util.xyz(x, y, 0)
+        for key, (x, y, z) in pos.items():
+            self.node[key]["position"] = util.xyz(x, y, z)
 
     def add_vops(self):
         """ Automatically add vops if they're not present """
diff --git a/abp/graphstate.py b/abp/graphstate.py
index 91d2c35..1498872 100644
--- a/abp/graphstate.py
+++ b/abp/graphstate.py
@@ -114,7 +114,7 @@ class GraphState(object):
         va = self.node[a]["vop"]
         vb = self.node[b]["vop"]
         new_edge, self.node[a]["vop"], self.node[b]["vop"] = \
-            clifford.cz_table[edge, va, vb]
+            clifford.cz_table[int(edge), va, vb]
         if new_edge != edge:
             self.toggle_edge(a, b)
 
diff --git a/demo.gif b/img/demo.gif
similarity index 100%
rename from demo.gif
rename to img/demo.gif
diff --git a/img/viz.png b/img/viz.png
new file mode 100644
index 0000000..1997da8
Binary files /dev/null and b/img/viz.png differ