Add correction code style with pep8