diff --git a/driver.py b/driver.py
index 7a77212..e75db46 100644
--- a/driver.py
+++ b/driver.py
@@ -3,7 +3,7 @@
 import numpy as np, os, sys
 from scipy.io import loadmat
 from run_12ECG_classifier import load_12ECG_model, run_12ECG_classifier
-
+from tqdm import tqdm
 def load_challenge_data(filename):
 
     x = loadmat(filename)
@@ -61,8 +61,7 @@ def save_challenge_predictions(output_directory,filename,scores,labels,classes):
     # Iterate over files.
     print('Extracting 12ECG features...')
     num_files = len(input_files)
-
-    for i, f in enumerate(input_files):
+    for i, f in tqdm(enumerate(input_files)):
         print('    {}/{}...'.format(i+1, num_files))
         tmp_input_file = os.path.join(input_directory,f)
         data,header_data = load_challenge_data(tmp_input_file)