Modernize and flake8 the package