diff --git a/src/pyplaml/diagram_layout.py b/src/pyplaml/diagram_layout.py
index e0f1df0a1dff3e5725d9774f8ea6357fc419e1f3..e54894208757e33a6234a1fd00823b6acb1d93c7 100644
--- a/src/pyplaml/diagram_layout.py
+++ b/src/pyplaml/diagram_layout.py
@@ -9,13 +9,18 @@ from pyplaml import *
 
 
 class DiagramLayout(ABC):
+    """Base class for layouts which define the position of objects in a diagram."""
+
     objects: typing.Dict[str, DiagramObject]
 
     @abstractmethod
     def apply(self, objects: typing.Dict[str, DiagramObject], scale_x: float = 1, scale_y: float = 1):
+        """Applies layout for provided objects."""
         self.objects = objects
 
     def get_graph(self) -> nx.DiGraph:
+        """Converts provided objects to a directed graph."""
+
         g = nx.DiGraph()
         for name, o in self.objects.items():
             g.add_node(o.get_key())