diff --git a/screening/etc/data-set-handler.py b/screening/etc/data-set-handler.py
index 90aaa319bfdbb416d78a269e0f93ff2c61cb92a6..954d313f3bf521b15e11415e7f721b795426482a 100755
--- a/screening/etc/data-set-handler.py
+++ b/screening/etc/data-set-handler.py
@@ -35,6 +35,8 @@ class ImageDataSetFlexible(SimpleImageDataConfig):
         return image_tokens
 
     """
+    Overrides the default implementation which returns (1, maxTileNumber) geometry.
+    
     Calculates the width and height of the matrix of tiles (a.k.a. fields or sides) in the well.
     
     Parameter imageMetadataList: a list of metadata for each encountered image
@@ -46,6 +48,8 @@ class ImageDataSetFlexible(SimpleImageDataConfig):
         return Geometry.createFromRowColDimensions(maxTileNumber / 3, 3);
     
     """
+    Overrides the default implementation which does the same thing (to demonstrate how this can be done). 
+    
     For a given tile number and tiles geometry returns (x,y) which describes where the tile is
     located on the well.