We should probably import `pyplot` (e.g., `import matplotlib.pyplot as plt`) instead of `pylab`, as the latter is really meant for interactive work (see http://matplotlib.org/faq/usage_faq.html#matplotlib-pylab-and-pyplot-how-are-they-related).