diff --git a/code/chirpdetection.py b/code/chirpdetection.py
index aa0d0fc..2373744 100644
--- a/code/chirpdetection.py
+++ b/code/chirpdetection.py
@@ -217,8 +217,6 @@ def main(datapath: str) -> None:
         # iterate through all fish
         for i, track_id in enumerate(np.unique(ident[~np.isnan(ident)])[:2]):
 
-            # <------------------------------------------ Find best electrodes here
-            # <------------------------------------------ Iterate through electrodes
             # get indices for time array in time window
             window_index = np.arange(len(idx))[
                 (ident == track_id) & (time[idx] >= t0) & (
@@ -240,6 +238,8 @@ def main(datapath: str) -> None:
             electrode = np.argsort(np.nanmean(powers_temp, axis=0))[-1]
             # electrode = best_electrodes[0]
 
+            # <------------------------------------------ Iterate through electrodes
+
             # plot spectrogram
             plot_spectrogram(axs[0, i], data_oi[:, electrode], data.samplerate)