Fix docstrings. PEP8 compliance.

Making doctests for long output pass is tricky, see e.g.:
http://stackoverflow.com/questions/13395610/