fixing comments and flake8