Missing `import matplotlib.pyplot as plt`