diff --git a/R/getStatsForTrees.R b/R/getStatsForTrees.R
index 9f6261b..f6e3bf8 100644
--- a/R/getStatsForTrees.R
+++ b/R/getStatsForTrees.R
@@ -55,7 +55,7 @@ getStatsForTrees = function(trees, nodes.train, type = "binary", base_score = 0.
         rightG = tree[ID==right,G]
         
         tree[r,G:=leftG+rightG]
-        w=tree[r,-G/H]
+        w=tree[r,-G/H] # w=tree[r,-G/(H+lambda)]
         
         tree[r,weight:=w]
         tree[ID==left,previous_weight:=w]
@@ -67,4 +67,4 @@ getStatsForTrees = function(trees, nodes.train, type = "binary", base_score = 0.
   }
   
   return (tree_list)
-}
\ No newline at end of file
+}