pep8 fixes
minor tidy up.