Fix for PEP8