Make code PEP8 clean