Fix PEP8 warnings/