diff --git a/sCOCA_ML/models/base_class_models.py b/sCOCA_ML/models/base_class_models.py
index 3ce66e6..ba62a91 100644
--- a/sCOCA_ML/models/base_class_models.py
+++ b/sCOCA_ML/models/base_class_models.py
@@ -1,3 +1,7 @@
+"""
+Base class for all models in the sCOCA_ML package.
+"""
+
 import torch