Adding the improved benchmark code
Adding the changed benchmark code