diff --git a/scripts/checkTrackSelection.py b/scripts/checkTrackSelection.py
index 488dd7d..7915a4d 100644
--- a/scripts/checkTrackSelection.py
+++ b/scripts/checkTrackSelection.py
@@ -3,8 +3,8 @@
 # @Author: Elena Graverini
 # @Date:   2015-10-27 18:26:49
 # @Last Modified by:   Elena Graverini
-# @Last Modified time: 2017-05-03 18:58:44
-
+# @Last Modified time: 2017-05-03 19:03:41
+import sys
 import os
 from math import sqrt
 import ROOT as r
@@ -24,7 +24,7 @@
 val = {'TT': 5, 'IT': 7}
 layer = {'TT': 'TTaU', 'IT': 'T3X2'}
 cut = {'TT': 12, 'IT': 20}
-stat = 10000000
+stat = 20000000
 
 r.gROOT.SetBatch(True)
 r.gROOT.ProcessLine('.X %s/../include/lhcbstyle.C' % os.getcwd())
@@ -46,26 +46,27 @@
 
 def s_over_b(s, b):
     h = s.Clone()
-    res = h.Divide(b)
-    res.name = 's_over_b'
-    return res
+    h.Divide(b)
+    h.name = 's_over_b'
+    return h
 
 
 def s_over_sqrtb(s, b):
     h = s.Clone()
     sqrtb = sqrt_of_2d_histogram(b)
-    res = h.Divide(sqrtb)
-    res.name = 's_over_sqrtb'
-    return res
+    h.Divide(sqrtb)
+    h.name = 's_over_sqrtb'
+    return h
 
 
 def punzi_significance(s, b):
     h = s.Clone()
-    sb = s.Add(b)
+    sb = s.Clone()
+    sb.Add(b)
     sqrt_sb = sqrt_of_2d_histogram(sb)
-    res = h.Divide(sqrt_sb)
-    res.name = 'punzi_significance'
-    return res
+    h.Divide(sqrt_sb)
+    h.name = 'punzi_significance'
+    return h
 
 
 def put_fill(c, fill):
@@ -206,7 +207,8 @@
             t_sig = f.Get('STADCTrackMonitor/HitInfo/%s' % layer[det])
 
             f, t_sig = printSelection(f, t_sig, det, fill, stat)
-            for step in [0, 12, 48, 54, 60]:
-                f, t_sig = printADC(f, t_sig, det, fill, stat, step)
+            if '--steps' in sys.argv:
+                for step in [0, 12, 48, 54, 60]:
+                    f, t_sig = printADC(f, t_sig, det, fill, stat, step)
 
             f.Close()