minor fix
more updates to benchmark code