Better benchmark code