Fixing PEP 8