diff --git a/code/read_baseline_data.py b/code/read_baseline_data.py
index a0cf727..eca0157 100644
--- a/code/read_baseline_data.py
+++ b/code/read_baseline_data.py
@@ -5,8 +5,6 @@ from IPython import embed
 def read_baseline_eod(dataset):
     base = dataset.split(os.path.sep)[-1] + ".nix"
     nix_file = nix.File.open(os.path.join(dataset, base))
-    embed()
-    nix_file.close()
     b = nix_file.blocks[0]
     for t in b.tags():
         if "baseline" in t.name:
@@ -15,11 +13,11 @@ def read_baseline_eod(dataset):
     eod = t.retrieve_data("LocalEOD-1")
     time = eod_da.dimensions[0].axis(len(eod))
     nix_file.close()
-
+    return time, eod
     
 if __name__ == "__main__":
     data_dir = "../data"
     dataset = "2018-11-09-ad-invivo-1"
-    eod = read_baseline_eod(os.path.join(data_dir, dataset))
+    time, eod = read_baseline_eod(os.path.join(data_dir, dataset))