Fixing for PEP 8