- Improvements to benchmark script