diff --git a/ephys_analysis.py b/ephys_analysis.py
index 9c560f1..a579d30 100644
--- a/ephys_analysis.py
+++ b/ephys_analysis.py
@@ -17,8 +17,8 @@ def analyze_sams(filename):
     rtag_data.metadata.pprint(max_depth=-1)
     print(40*'*')
     stim_tag = g.multi_tags[0]
-    stim_pos = stim_tag.positions[:]
-    stim_extent = stim_tag.extents[:]
+    stim_pos = stim_tag.positions[:] # beginnings of stimulations
+    stim_extent = stim_tag.extents[:] # duration of stimulations
 
     for r in rtag_data.references:
         print(r.name, r.type)
@@ -27,24 +27,26 @@ def analyze_sams(filename):
 
     spike_data = []
 
-    for i in range(len(stim_pos)):
-        spike_data.append(stim_tag.retrieve_data(i, 'Spikes-1')[:])
+    for idx in range(len(stim_pos)):
+        spike_data.append(stim_tag.retrieve_data(idx, 'Spikes-1')[:])
 
 
-    dims = [[]] * len(voltage_trace)
-    for j in range(len(spike_data)):
-        dims[j] = voltage_trace.dimensions[j]
-    print(spike_data, len(spike_data))
+    # dims = [[]] * len(spike_data)
+    # for j in range(len(spike_data)):
+    #     dims[j] = spike_data.dimensions[j]
+    # print(spike_data, len(spike_data))
+    # embed()
+    # exit()
     f.close()
-    return stim_pos, stim_extent, spike_data, voltage_trace, dims
+    return stim_pos, stim_extent, spike_data, voltage_trace
 
 
 if __name__ == '__main__':
     data_dir = '/home/lisa/data'
     data_set = '2019-06-24-aa-invivo-1'
-    stim_pos, stim_extent, spike_data, voltage_trace, dims = analyze_sams(os.path.join(data_dir, data_set, data_set + '.nix'))
+    stim_pos, stim_extent, spike_data, voltage_trace = analyze_sams(os.path.join(data_dir, data_set, data_set + '.nix'))
     for i in range(len(stim_pos)):
-        plt.eventplot(dims[i], spike_data[i])
+        plt.scatter(voltage_trace.dimensions[i].axis(1000), spike_data[i][:1000])
         plt.show()
         # spikes = stim_tag.retrieve_data(i, 'Spikes-1')[:]
         # print(spikes)