The matplotlib import can be removed.