Fixing PEP8 whitespace