improvements in benchmark.py script