Add new parameter to bench.py