diff --git a/modules/optimizer.py b/modules/optimizer.py
index e217c0c36aa6c3f30f41c99e74ef8ff1e957333b..e204df3181053c9fa9e5b85db27cdf39c7b36380 100644
--- a/modules/optimizer.py
+++ b/modules/optimizer.py
@@ -97,10 +97,10 @@ class Optimizer(object):
                 def sigmoid(number):
                     return 1/ (1 + torch.exp(-number))
 
-                abs_sigm = sigmoid(scale_factor*(abs_parameters))
-                cost_abs = abs_sigm * 25 + (1 - abs_sigm) * 4.166
+                abs_sigm = sigmoid(scale_factor*(abs_parameters))  # Scaled sigmoid function
+                cost_abs = abs_sigm * 25 + (1 - abs_sigm) * 4.166   # 25 is the cost of Pb and 4.166 for Fe
                 scint_sigm = sigmoid(scale_factor*(scint_parameters))
-                cost_scint = scint_sigm * 2500 + (1 - scint_sigm) * 0.01
+                cost_scint = scint_sigm * 2500 + (1 - scint_sigm) * 0.01 # 2500 is the cost of PbWO4 and 0.01 for Polystyrene. 
 
                 combined = torch.cat((cost_abs, cost_scint), dim=0)