Pep 8 changes