name 'plt' is not defined after download helper_functions.py

[yanliu1111]

After download helper_functions.py, plot decision boundary of the model, got the error for plt undefined
NameError Traceback (most recent call last)
in <cell line: 2>()
1 # Plot decision boundary of the model
----> 2 plt.figure(figsize=(12, 6))
3 plt.subplot(1, 2, 1)
4 plt.title("Train")
5 plot_decision_boundary(model_0, X_train, y_train)

NameError: name 'plt' is not defined

[yanliu1111]

I forgot to rerun this
import matplotlib.pyplot as plt