Updated benchmarking script