Fix PEP8 stuff