diff --git a/deepsecure/catted_generator.py b/deepsecure/catted_generator.py
index 6eb8d14..35db7a4 100644
--- a/deepsecure/catted_generator.py
+++ b/deepsecure/catted_generator.py
@@ -152,8 +152,11 @@ def __init__(self, num_encoder_layers, fix_encoder, tagged):
             # Decoder for 5-layer encoder (64*2=128 input channels after concatenation)
             decoder_lis = [
                 # Process concatenated features
+                # First ResNet block to process the concatenated features (128 channels)
                 ResnetBlock(64*2),
+                # Second ResNet block for further feature transformation
                 ResnetBlock(64*2),
+                # Third ResNet block for deeper feature refinement
                 ResnetBlock(64*2),
                 nn.UpsamplingNearest2d(scale_factor=2),