small pep 8 changes