diff --git a/pyrelacs/ui/mainwindow.py b/pyrelacs/ui/mainwindow.py
index 153f1ef..548856e 100644
--- a/pyrelacs/ui/mainwindow.py
+++ b/pyrelacs/ui/mainwindow.py
@@ -36,9 +36,15 @@ class PyRelacs(QMainWindow):
             Qt.ToolButtonStyle.ToolButtonTextBesideIcon
         )  # Ensure icons are displayed with text
         self.setWindowTitle("PyRelacs")
+
         self.beat_plot = pg.PlotWidget()
         self.power_plot = pg.PlotWidget()
 
+        self.beat_plot.addLegend()
+        self.power_plot.addLegend()
+        self.beat_plot.setBackground("w")
+        self.power_plot.setBackground("w")
+
         self.threadpool = QThreadPool()
         self.repros = Repro()
 
@@ -60,8 +66,8 @@ class PyRelacs(QMainWindow):
         widget.setLayout(layout)
         self.setCentralWidget(widget)
 
-        filename = path.joinpath(path.cwd(), "data")
-        self.nix_file = nix.File.open(str(filename), nix.FileMode.Overwrite)
+        filename = path.joinpath(path.cwd(), "data.nix")
+        self.nix_file = nix.File.open(str(filename), nix.FileMode.ReadWrite)
 
     def create_actions(self):
         self._rlx_exitaction = QAction(
@@ -216,15 +222,15 @@ class PyRelacs(QMainWindow):
 
             f_sq, powerspec_sq = welch(beat_squared, fs=40_000.0)
             powerspec_sq = decibel(powerspec_sq)
-            peaks = find_peaks(powerspec_sq, prominence=20)[0]
+            peaks = find_peaks(powerspec_sq, prominence=10)[0]
             pen = pg.mkPen(colors[i])
             self.beat_plot.plot(
                 np.arange(0, len(beat)) / 40_000.0,
                 beat_squared,
                 pen=pen,
-                # name=stim.name,
+                name=stim.name,
             )
-            self.power_plot.plot(f_sq, powerspec_sq, pen=pen)
+            self.power_plot.plot(f_sq, powerspec_sq, pen=pen, name=stim.name)
             self.power_plot.plot(f[peaks], powerspec_sq[peaks], pen=None, symbol="x")
 
     def connect_dac(self):