Change benchmark.py