update bench.py to latest implementation