diff --git a/simulations/airshower_beacon_simulation/ca_period_from_shower.py b/simulations/airshower_beacon_simulation/ca_period_from_shower.py
index f6c0dd3..73ab606 100755
--- a/simulations/airshower_beacon_simulation/ca_period_from_shower.py
+++ b/simulations/airshower_beacon_simulation/ca_period_from_shower.py
@@ -158,6 +158,7 @@ if __name__ == "__main__":
     Xref = 400
 
     N_runs = 3
+    remove_beacon_from_trace = False
 
     ####
     fname_dir = path.dirname(fname)
@@ -202,7 +203,7 @@ if __name__ == "__main__":
         ev.antennas[i].orig_t = ev.antennas[i].t
         ev.antennas[i].t += clock_phase_time
 
-        if False: # remove beacon from trace
+        if remove_beacon_from_trace:
             meas_phase = ant.beacon_info[freq_name]['phase']
             f = ant.beacon_info[freq_name]['freq']
             ampl = ant.beacon_info[freq_name]['amplitude']
@@ -253,7 +254,6 @@ if __name__ == "__main__":
     scale4d = dXref*np.tan(np.deg2rad(4.))
 
     if not True: #quicky
-        N_runs = 2
         x_coarse = np.linspace(-scale2d, scale2d, 4)
         y_coarse = np.linspace(-scale2d, scale2d, 4)
 
@@ -364,7 +364,7 @@ if __name__ == "__main__":
             _, __, p, ___ = rit.shower_plane_slice(ev, X=Xref, Nx=len(x), Ny=len(y), wx=scale2d, wy=scale2d, xoff=xoff, yoff=yoff, zgr=0)
 
             fig, axs = plt.subplots()
-            axs.set_title(f"Shower reconstruction with best k, Grid Run {r}")
+            axs.set_title(f"Shower slice for best k, Grid Run {r}")
             axs.set_ylabel("vxvxB [km]")
             axs.set_xlabel("  vxB [km]")
             axs.set_aspect('equal', 'datalim')