PEP8 compliance edit