Pep 8 fixed