Fix style according to pep8