Updated a few lines, to comply with flake8.