Index: trunk/MagicSoft/Mars/mjtrain/MJTrainSeparation.cc
===================================================================
--- trunk/MagicSoft/Mars/mjtrain/MJTrainSeparation.cc	(revision 7670)
+++ trunk/MagicSoft/Mars/mjtrain/MJTrainSeparation.cc	(revision 7671)
@@ -165,4 +165,6 @@
     gPad->SetFrameBorderMode(0);
     gPad->SetLogx();
+    gPad->SetGridx();
+    gPad->SetGridy();
     h.DrawCopy();
     g.DrawCopy("same");
@@ -176,4 +178,6 @@
     gPad->SetBorderMode(0);
     gPad->SetFrameBorderMode(0);
+    gPad->SetGridx();
+    gPad->SetGridy();
     gr4.SetMarkerStyle(kFullDotMedium);
     gr4.DrawClone("ALP")->SetBit(kCanDelete);
@@ -186,4 +190,6 @@
     gPad->SetFrameBorderMode(0);
     gPad->SetLogx();
+    gPad->SetGridx();
+    gPad->SetGridy();
     gPad->AddExec("color", "gStyle->SetPalette(1, 0);");
     res1.DrawCopy("colz");
@@ -193,4 +199,6 @@
     gPad->SetFrameBorderMode(0);
     gPad->SetLogx();
+    gPad->SetGridx();
+    gPad->SetGridy();
     gPad->AddExec("color", "gStyle->SetPalette(1, 0);");
     res2.DrawCopy("colz");
Index: trunk/MagicSoft/Mars/mjtrain/MJTrainSeparation.h
===================================================================
--- trunk/MagicSoft/Mars/mjtrain/MJTrainSeparation.h	(revision 7670)
+++ trunk/MagicSoft/Mars/mjtrain/MJTrainSeparation.h	(revision 7671)
@@ -32,5 +32,5 @@
     Double_t GetDataRate(MDataSet &set, Double_t &num) const;
     Double_t GetNumMC(MDataSet &set) const;
-    Bool_t   AutoTrain();
+    Bool_t   AutoTrain(MDataSet &set, UInt_t &on, UInt_t &off);
 
 public: