fixing code to compy with pep8