Simplify benchmark code