These new notebooks look great. I don't see this import used in this notebook, though – is it needed?

Added the missing legend. Someone superfluous in this example though. Thanks!