small pep 8 fixes