Fix PEP8 compliance