Clean up comment
Pep8 cleanup