Pep 8 fixes