diff --git a/code/read_chirp_data.py b/code/read_chirp_data.py
index 5c01e11..5067922 100644
--- a/code/read_chirp_data.py
+++ b/code/read_chirp_data.py
@@ -19,15 +19,17 @@ def load_chirp_spikes(dataset):
            df = l.split(":")[-1]
         if "contrast" in l and "true" not in l:
             contrast = l.split(":")[-1]
+        if "chirpsize" in l:
+            cs = l.split(":")[-1]
         if "#Key" in l:
-            spikes[(index, df, contrast)] = {}
+            spikes[(index, df, contrast, cs)] = {}
         if "chirp index" in l:
             ci = int(l.split(":")[-1])
         if "beat phase" in l:
             phase = float(l.split(":")[-1])
-            spikes[(index, df, contrast)][(ci, phase)] = []
+            spikes[(index, df, contrast, cs)][(ci, phase)] = []
         if len(l.strip()) != 0 and "#" not in l:
-            spikes[(index, df, contrast)][(ci, phase)].append(float(l))
+            spikes[(index, df, contrast, cs)][(ci, phase)].append(float(l))
     return spikes
 
 
@@ -47,12 +49,15 @@ def load_chirp_times(dataset):
             df = l.split(":")[-1]
         if "contrast" in l and "true" not in l:
             contrast = l.split(":")[-1]
+        if "chirpsize" in l:
+            cs = l.split(":")[-1]
         if "#Key" in l:
-            chirp_times[(index, df, contrast)] = []
+            chirp_times[(index, df, contrast, cs)] = []
         if len(l.strip()) != 0 and "#" not in l:
-            chirp_times[(index, df, contrast)].append(float(l.split()[1]))
+            chirp_times[(index, df, contrast, cs)].append(float(l.split()[1]))
     return chirp_times
 
+
 if __name__ == "__main__":
     data_dir = "../data"
     dataset = "2018-11-09-ad-invivo-1"