https://legendhjc.blogspot.com/2023/04/how-to-add-legend-in-matplotlib.html