Fix.

yafn/__main__.py

@@ -150,6 +150,9 @@ def distance(graph, start, end):
   visited = []
   queue = [[start]]
 
+  if start == end:
+    return 0
+
   while queue:
     path = queue.pop(0)
     node = path[-1]