diff --git a/groupConnectivityNaive.py b/groupConnectivityNaive.py
index af4c4bc734ef0d0aa455b6f84d6223364473f532..5bbd2fed41a301845b214a0670be0a65f0a94c5c 100644
--- a/groupConnectivityNaive.py
+++ b/groupConnectivityNaive.py
@@ -1,5 +1,12 @@
 from sage.graphs.graph import Graph, DiGraph
 
+def prettifyGraph(G_):
+  """Convert graph into directed one and number its edges."""
+
+  E = [ (u, v, i) for i, (u, v, _) in enumerate(G_.edges()) ]
+  return DiGraph([G_.vertices(), E], format='vertices_and_edges')
+
+
 def flowEnumerator(G, group):
   """Enumerate all flows of given graph.