better comments, pep8