Better TODOs.
pep8 fixes.