Fix Flake 8 errors