Changed the driver file to print time for prediction for each sample