updated benchmark.py