Fixed PEP8 problems