diff --git a/second_try.py b/second_try.py
index aaad48a..9d0de30 100644
--- a/second_try.py
+++ b/second_try.py
@@ -40,7 +40,7 @@ for infodataset in infodatasets:
     ID.append(identifier)
 
 
-for dataset in datasets:
+for idx, dataset in enumerate(datasets):
     #input of the function
     frequency, time, amplitude, eodf, deltaf, stimulusf, duration, pause = parse_dataset(dataset)
     dm = np.mean(duration)
@@ -62,7 +62,7 @@ for dataset in datasets:
     freq_all.append(norm.tolist())
     time_all.append(ct_arr)
 
-    plt.plot(ct_arr, norm, color = 'grey', label='fish=%s' % ID)
+    plt.plot(ct_arr, norm, color = 'grey', label='fish=%s' % ID[idx])
 
     # fit function
     sv, sc = curve_fit(step_response, ct_arr[ct_arr < dm], norm[ct_arr < dm]) #step_values and step_cov