Fix PEP8 errors in the code