You should also add the proper imports, so somebody can copy-paste this code and run it. IPython can run this code, even with the `>>>` and `...` characters on it.