diff --git a/scripts/mbedtls_dev/test_generation.py b/scripts/mbedtls_dev/test_generation.py
index 682f7b0365..f1b268229a 100644
--- a/scripts/mbedtls_dev/test_generation.py
+++ b/scripts/mbedtls_dev/test_generation.py
@@ -130,10 +130,12 @@ class BaseTarget(metaclass=ABCMeta):
 
 
 class TestGenerator:
-    """Generate test data."""
+    """Generate test cases and write to data files."""
     def __init__(self, options) -> None:
         self.test_suite_directory = getattr(options, 'directory')
-        # Add file Targets which have been declared in other modules
+        # Update `targets` with an entry for each child class of BaseTarget.
+        # Each entry represents a file generated by the BaseTarget framework,
+        # and enables generating the .data files using the CLI.
         self.targets.update({
             subclass.target_basename: subclass.generate_tests
             for subclass in BaseTarget.__subclasses__()