minor changes for pep8