or maybe `from matplotlib import pyplot as plt`?