PEP 8 fixes, comments.

Getting ready for a PR!