Fix PEP8 issues across the code