Remove unused imports: staticcheck warnings (pyflakes)