From 3c9bbb6a91939defaade786f4cfa468b05de7e38 Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 19 May 2023 08:17:31 +0200
Subject: [PATCH] bf

---
 trail_analysis.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/trail_analysis.py b/trail_analysis.py
index 27bd40a..d685736 100644
--- a/trail_analysis.py
+++ b/trail_analysis.py
@@ -205,6 +205,7 @@ class Trial(object):
 
             freq_range = (np.nanmin(self.fish_freq), np.nanmax(self.fish_freq))
             ax.set_ylim(freq_range[0] - 20, freq_range[1] + 10)
+        ax.legend(loc = 'upper right', bbox_to_anchor=(1, 1))
         ax.set_title(self.folder)
         plt.show()