diff --git a/src/agora/io/cells.py b/src/agora/io/cells.py
index a8ca61b0e13c28b266d5bb2b9bd8fd02384106e3..095b59c4d15448ca30c5a9786fdddcfb9733f95b 100644
--- a/src/agora/io/cells.py
+++ b/src/agora/io/cells.py
@@ -160,8 +160,11 @@ class Cells:
     def group_by_traps(
         self, traps: t.Collection, cell_labels: t.Collection
     ) -> t.Dict[int, t.List[int]]:
-        # returns a dict with traps as keys and list of labels as value
-        # Data is a
+        """
+        Returns a dict with traps as keys and list of labels as value.
+        Note that the total number of traps are calculated from Cells.traps.
+
+        """
         iterator = groupby(zip(traps, cell_labels), lambda x: x[0])
         d = {key: [x[1] for x in group] for key, group in iterator}
         d = {i: d.get(i, []) for i in self.traps}