diff --git a/pynode_next/edge.py b/pynode_next/edge.py
index e6752e2..c1e01df 100644
--- a/pynode_next/edge.py
+++ b/pynode_next/edge.py
@@ -49,6 +49,7 @@ def target(self, source=None):
     def set_attribute(self, name, value):
         """Sets an attribute of the edge"""
         self._attrs[name] = value
+        return self
 
     def attribute(self, name):
         """Gets an attribute of the edge"""
@@ -176,6 +177,7 @@ def traverse(self, initial_node=None, color=Color.RED, keep_path=True):
         else:
             # stores the changed color
             self._color = color
+        return self
 
     def __str__(self):
         return f"({self._source}, {self._target})"