pep8 + a little fix