When trying to run the first exercise, the plotting function produced an error:
AttributeError: 'Legend' object has no attribute 'legendHandles'
I believe this is because the code is using an older version of matplotlib. Using Google Colab, I tested this by adding the following code to the top got it to work:
pip install "numpy<2" "matplotlib==3.6"