Fixed PEP 8 issues