diff --git a/dhnx/optimization/optimization_models.py b/dhnx/optimization/optimization_models.py
index e5993c9a..7d6c87ef 100644
--- a/dhnx/optimization/optimization_models.py
+++ b/dhnx/optimization/optimization_models.py
@@ -8,6 +8,8 @@
 by the contributors recorded in the version control history of the file,
 available from its original location:
 
+TODO : Add class for the optimization of aggregated Networks
+
 SPDX-License-Identifier: MIT
 """