Pep 8 cleanup