Fix PEP8 lint problems