minor pep8 moidification