Add cli setup in `src/bench/__main__.py`
Add cli setup in
src/bench/__main__.py