diff --git a/benchmarks/track_finding/scripts/gen_multiple_tracks.cxx b/benchmarks/track_finding/scripts/gen_multiple_tracks.cxx
index 7960c8ca04b629c3c0ca0081d37041e0be8f3914..8596e929d190d7b023f37f3fbbf7da07b88508cb 100644
--- a/benchmarks/track_finding/scripts/gen_multiple_tracks.cxx
+++ b/benchmarks/track_finding/scripts/gen_multiple_tracks.cxx
@@ -44,7 +44,7 @@ void gen_multiple_tracks(int n_events = 100,
       // Define momentum
       Double_t p     = r1->Uniform(1.0, 10.0);
       Double_t phi   = r1->Uniform(0.0, 2.0 * M_PI);
-      Double_t costh = r1->Uniform(-0.05, 0.05); //r1->Uniform(cos_theta_min, cos_theta_max);
+      Double_t costh = r1->Uniform(cos_theta_min, cos_theta_max);
       Double_t th    = std::acos(costh);
       Double_t px    = p * std::cos(phi) * std::sin(th);
       Double_t py    = p * std::sin(phi) * std::sin(th);
diff --git a/benchmarks/track_finding/scripts/rec_multiple_tracks.cxx b/benchmarks/track_finding/scripts/rec_multiple_tracks.cxx
index 788b9e2a54964b742b6000b5553b4fdaddf79fb5..4a8429849dcc1bf064295636deb05369a2e02a42 100644
--- a/benchmarks/track_finding/scripts/rec_multiple_tracks.cxx
+++ b/benchmarks/track_finding/scripts/rec_multiple_tracks.cxx
@@ -102,6 +102,9 @@ int rec_multiple_tracks(const char* fname = "topside/rec_multiple_tracks.root")
                  .Define("N_Hits",       [](std::vector<eic::TrackerHitData> hits) { return hits.size();}, {"trackingHits"})
                  .Define("N_BarrelHits", [](std::vector<eic::TrackerHitData> hits) { return hits.size();}, {"TrackerBarrelRecHits"})
                  .Define("N_EndcapHits", [](std::vector<eic::TrackerHitData> hits) { return hits.size();}, {"TrackerEndcapRecHits"})
+                 .Define("N_BarrelTOFHits", [](std::vector<eic::TrackerHitData> hits) { return hits.size();}, {"TrackerBarrelTOFRecHits"})
+                 .Define("N_ForwardTOFHits", [](std::vector<eic::TrackerHitData> hits) { return hits.size();}, {"TrackerForwardTOFRecHits"})
+                 .Define("N_BackwardTOFHits", [](std::vector<eic::TrackerHitData> hits) { return hits.size();}, {"TrackerBackwardTOFRecHits"})
                  ;
 
   auto h_nTracks_vs_theta = df0.Histo2D({"h_nTracks_vs_theta", "; #theta; N tracks ", 40,0,180,10, 0, 10}, "theta0","nTracks");
@@ -120,6 +123,9 @@ int rec_multiple_tracks(const char* fname = "topside/rec_multiple_tracks.root")
   auto hNhits_vs_theta = df0.Histo1D({"hNhits_vs_theta", "; #theta [deg.]",   40, 0, 180 }, "theta0", "N_Hits");
   auto hBarrel_N_vs_theta = df0.Histo1D({"hBarrel_N_vs_theta", "; #theta [deg.]",   40, 0, 180 }, "theta0", "N_BarrelHits");
   auto hEndcap_N_vs_theta = df0.Histo1D({"hEndcap_N_vs_theta", "; #theta [deg.]",   40, 0, 180 }, "theta0", "N_EndcapHits");
+    auto hBarrelTOF_N_vs_theta = df0.Histo1D({"hBarrelTOF_N_vs_theta", "; #theta [deg.]",   40, 0, 180 }, "theta0", "N_BarrelTOFHits");
+    auto hForwardTOF_N_vs_theta = df0.Histo1D({"hForwardTOF_N_vs_theta", "; #theta [deg.]",   40, 0, 180 }, "theta0", "N_ForwardTOFHits");
+    auto hBackwardTOF_N_vs_theta = df0.Histo1D({"hBackwardTOF_N_vs_theta", "; #theta [deg.]",   40, 0, 180 }, "theta0", "N_BackwardTOFHits");
 
   auto hHits_Nhits  = df0.Histo1D({"hHits_Nhits", "; #theta [deg.]",       20, 0, 20 }, "N_Hits");
   auto hBarrel_Nhits  = df0.Histo1D({"hBarrel_Nhits", "; #theta [deg.]",   20, 0, 20 }, "N_BarrelHits");