Flake8 fixes for the library