Draw return doc

Went to look up the return types of the `draw_*` functions recently and was annoyed that they were not documented.

This PR also includes PEP8 clean up of `draw_pylab.py`