diff --git a/speed_benchmark/README.md b/speed_benchmark/README.md
index 80cbeee..bb33b5b 100644
--- a/speed_benchmark/README.md
+++ b/speed_benchmark/README.md
@@ -1,4 +1,4 @@
 Steps:
 
-1. Run `python speed_benchmark/speed_benchmark.py` which times operations and writes their time to `speed_benchmark/info_a100_py2.jsonl` (feel free to change the name for your profiling).
-2. Run `python speed_benchmark/make_plot_with_jsonl.py`, which produces the `speed_benchmark/plot_with_info.pdf`.
\ No newline at end of file
+1. Run `python speed_benchmark/speed_benchmark.py` which times operations and writes their time to `speed_benchmark/info_a100_py2.jsonl` (change the name of the jsonl to a different name for your profiling).
+2. Run `python speed_benchmark/make_plot_with_jsonl.py`, which produces the `speed_benchmark/plot_with_info.pdf`. Again make sure you change the jsonl which is being processed.
\ No newline at end of file
diff --git a/speed_benchmark/make_plot_with_jsonl.py b/speed_benchmark/make_plot_with_jsonl.py
index 0920851..8897564 100644
--- a/speed_benchmark/make_plot_with_jsonl.py
+++ b/speed_benchmark/make_plot_with_jsonl.py
@@ -20,6 +20,7 @@ if __name__ == '__main__':
 
     ax = fig.add_subplot(gs[0, 0])
 
+    # TODO: change this to what you want.
     rdf = pd.read_json('speed_benchmark/info_a100_py2.jsonl', lines=True)
     df = rdf[rdf.batch_size == batch_size_for_plot1]
 
diff --git a/speed_benchmark/speed_benchmark.py b/speed_benchmark/speed_benchmark.py
index bd8a6e5..eccc455 100644
--- a/speed_benchmark/speed_benchmark.py
+++ b/speed_benchmark/speed_benchmark.py
@@ -97,5 +97,6 @@ if __name__ == '__main__':
 
                 info_json = json.dumps(info)
 
-                with open("speed_benchmark/info_a100_py2.jsonl", "a") as file:
+                # TODO: change this to what you want.
+                with open("speed_benchmark/info.jsonl", "a") as file:
                     file.write(info_json + "\n")