Add a Legend

Both lines are already labelled. Add plt.legend() to make the labels appear on the plot.
Expected:
matplotlib expected figure
Output:
Python
Setting up Python environment...
Output