Small code formatting change
pep8 fix