minor improvements to benchmark