Fixed issue with PEP8 compliance