improvements in benchmark.py script
improvements in benchmark.py script