Addresses @Jorge-C's comment regarding matplotlib.pyplot import.