Add flake8 check & resolve violations