diff --git a/deepsecure/cat_adv_image.py b/deepsecure/cat_adv_image.py
index 052ccb6..1250767 100644
--- a/deepsecure/cat_adv_image.py
+++ b/deepsecure/cat_adv_image.py
@@ -145,7 +145,9 @@ def __init__(self, device, model_extractor, generator, reg_g):
             generator: Main generator model
             reg_g: Regular generator for noise generation (False if not used)
         """
+        # Store the device (CPU/GPU) on which the model will operate
         self.device = device
+        # Assign the feature extraction model for processing input features
         self.model_extractor = model_extractor
         self.generator = generator
         self.box_min = cfg.BOX_MIN  # Minimum pixel value