Index: trunk/MagicSoft/Mars/mjoptim/MJOptimizeBase.h
===================================================================
--- trunk/MagicSoft/Mars/mjoptim/MJOptimizeBase.h	(revision 8679)
+++ trunk/MagicSoft/Mars/mjoptim/MJOptimizeBase.h	(revision 8698)
@@ -13,13 +13,15 @@
 protected:
     Int_t  fDebug;
-    Bool_t fEnableWeights;
+    Bool_t fEnableWeights;  // Enable using weights
 
-    TList fRules;
+    TList fRules;           // Contains the parameters which can be accessed by matrix column
+    TList fTrainParameters; // Parameters in the last columns
 
-    TList fPreCuts;
-    TList fTrainCuts;
-    TList fTestCuts;
-    TList fPreTasks;
-    TList fPostTasks;
+    TList fPreCuts;         // Cuts executed for training and testing
+    TList fTrainCuts;       // Cuts executed only in training
+    TList fTestCuts;        // Cuts executed only in testing
+    TList fPreTasks;        // Tasks executed before cut execution
+    TList fPostTasks;       // Tasks executed after cut execution
+    TList fTestTasks;       // Tasks executed after cut execution for testing
 
     void AddCut(TList &l, const char *rule);
@@ -39,4 +41,8 @@
     void AddPostTask(const char *rule,
                      const char *name="MWeight") { AddPar(fPostTasks, rule, name); }
+
+    void AddTestTask(MTask *t)                   { Add(fTestTasks, t); }
+    void AddTestTask(const char *rule,
+                     const char *name="MWeight") { AddPar(fTestTasks, rule, name); }
 
     void SetDebug(Bool_t b=kTRUE)      { fDebug = b; }