From dbaef21bf5a2579ce5b2ce0ff420c391350a8ceb Mon Sep 17 00:00:00 2001
From: Michele Ceriotti <michele.ceriotti@gmail.com>
Date: Mon, 14 Oct 2024 15:10:18 +0200
Subject: [PATCH] Legend!

---
 examples/thermostats/thermostats.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/thermostats/thermostats.py b/examples/thermostats/thermostats.py
index 2a282aa2..57f4784e 100644
--- a/examples/thermostats/thermostats.py
+++ b/examples/thermostats/thermostats.py
@@ -223,6 +223,7 @@
 ax.plot(output_data["time"], output_data["temperature"], "k-", label="All atoms")
 ax.set_xlabel(r"$t$ / ps")
 ax.set_ylabel(r"$\tilde{T}$ / K")
+ax.legend()
 plt.show()
 
 # %%