Clean up file to PEP8