pep8 fixes (thanks @justinvdm)