Correct code style

Also replace `# flake8: noqa` by `# noqa`