small pep 8 fixing