diff --git a/electron_main.py b/electron_main.py
index cffb246..5450576 100644
--- a/electron_main.py
+++ b/electron_main.py
@@ -171,10 +171,6 @@ if args.matching_weights and args.residuals:
 
 file_name = "seed"
 
-input_files = {}
-input_files["seed"] = "data/ghost_data_B_default_only_e_as_seed.root"
-input_files["def"] = "data/ghost_data_B_default_thesis.root"
-
 tree_names = {}
 tree_names["seed"] = "PrMatchNN_b60a058d.PrMCDebugMatchToolNN/MVAInputAndOutput"
 tree_names["def"] = "PrMatchNN.PrMCDebugMatchToolNN/MVAInputAndOutput"
@@ -183,15 +179,16 @@ if args.matching_weights and not args.residuals:
     os.chdir(os.path.dirname(os.path.realpath(__file__)))
     train_matching_ghost_mlp(
         prepare_data=args.prepare,
-        input_file="data/ghost_data_B_default_thesis.root",
-        tree_name=tree_names[file_name],  # B: 3e224c41
-        exclude_electrons=True,
-        only_electrons=False,
+        input_file="data/ghost_data_B_vars_thesis.root",
+        tree_name=tree_names[file_name],
+        exclude_electrons=False,
+        only_electrons=True,
+        filter_seeds=True,
         outdir="nn_electron_training",
-        n_train_signal=150e3,
-        n_train_bkg=150e3,
-        n_test_signal=20e3,
-        n_test_bkg=20e3,
+        n_train_signal=100e3,
+        n_train_bkg=100e3,
+        n_test_signal=10e3,
+        n_test_bkg=10e3,
     )
     # this ensures that the directory is correct
     os.chdir(os.path.dirname(os.path.realpath(__file__)))
diff --git a/moore_options/get_ghost_data.py b/moore_options/get_ghost_data.py
index e8f1fc5..6e98d50 100644
--- a/moore_options/get_ghost_data.py
+++ b/moore_options/get_ghost_data.py
@@ -34,7 +34,7 @@ import glob
 options.evt_max = -1
 
 decay = "B"  # D, B
-options.ntuple_file = f"data/ghost_data_{decay}_default_thesis.root"
+options.ntuple_file = f"data/ghost_data_{decay}_vars_thesis.root"
 
 options.input_type = "ROOT"
 if decay == "B":
diff --git a/moore_options/get_tracking_losses.py b/moore_options/get_tracking_losses.py
index 8d437de..71b9eea 100644
--- a/moore_options/get_tracking_losses.py
+++ b/moore_options/get_tracking_losses.py
@@ -38,7 +38,7 @@ tested by mc_matching_example.py
 
 """
 
-decay = "B"
+decay = "test"
 
 options.evt_max = -1
 
@@ -61,7 +61,7 @@ options.dddb_tag = "dddb-20210617"
 options.simulation = True
 options.input_type = "ROOT"
 
-options.ntuple_file = f"data/tracking_losses_ntuple_{decay}_zmag.root"
+options.ntuple_file = f"data/tracking_losses_ntuple_{decay}_endVelo2endT.root"
 
 
 def run_tracking_losses():
diff --git a/parameterisations/train_matching_ghost_mlps_electron.py b/parameterisations/train_matching_ghost_mlps_electron.py
index e5586e6..9752539 100644
--- a/parameterisations/train_matching_ghost_mlps_electron.py
+++ b/parameterisations/train_matching_ghost_mlps_electron.py
@@ -11,6 +11,7 @@ def train_matching_ghost_mlp(
     tree_name: str = "PrMatchNN.PrMCDebugMatchToolNN/MVAInputAndOutput",
     exclude_electrons: bool = False,
     only_electrons: bool = True,
+    filter_seeds: bool = False,
     n_train_signal: int = 50e3,  # 50e3
     n_train_bkg: int = 50e3,  # 500e3
     n_test_signal: int = 10e3,
@@ -54,10 +55,16 @@ def train_matching_ghost_mlp(
                     "abs(quality) > 0",
                     "Signal is defined as non-zero label",
                 )
-            rdf_bkg = rdf.Filter(
-                "quality == 0",
-                "Ghosts are defined as zero label",
-            )
+            if filter_seeds:
+                rdf_bkg = rdf.Filter(
+                    "quality == 0 && scifi_isElectron == 1",
+                    "Ghosts are defined as zero label",
+                )
+            else:
+                rdf_bkg = rdf.Filter(
+                    "quality == 0",
+                    "Ghosts are defined as zero label",
+                )
 
         rdf_signal.Snapshot(
             "Signal",
@@ -98,8 +105,13 @@ def train_matching_ghost_mlp(
     dataloader.AddVariable("distX", "F")
     dataloader.AddVariable("distY", "F")
     dataloader.AddVariable("dSlope", "F")
-    dataloader.AddVariable("dSlopeY", "F")
-    # dataloader.AddVariable("zMag", "F")
+    # dataloader.AddVariable("dSlopeY", "F")
+    # dataloader.AddVariable("zmag", "F")
+    dataloader.AddVariable("eta", "F")
+    # dataloader.AddVariable("dEta", "F")
+    # dataloader.AddVariable("eta_scifi", "F")
+
+
 
     dataloader.AddSignalTree(signal_tree, 1.0)
     dataloader.AddBackgroundTree(bkg_tree, 1.0)