diff --git a/src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/__init__.py b/src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/__init__.py
new file mode 100644
index 0000000..89070de
--- /dev/null
+++ b/src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/__init__.py
@@ -0,0 +1 @@
+from .core import main
\ No newline at end of file
diff --git a/src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/main.py b/src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/core.py
similarity index 94%
rename from src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/main.py
rename to src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/core.py
index a143b86..dedf263 100644
--- a/src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/main.py
+++ b/src/aarch64-linux-flush-dcache/visualizer/ftrace2pyplot/core.py
@@ -10,7 +10,7 @@ import numpy as np
 import seaborn as sns
 import matplotlib.pyplot as plt
 
-from arguments import parser
+from .arguments import parser
 
 def parse_basename_into_byte_count(basename: str) -> int:
     ret: int = -1
@@ -20,9 +20,9 @@ def parse_basename_into_byte_count(basename: str) -> int:
     if pat is not None:
         size = int(pat.group("size"))
         match pat.group("unit"):
-            case "K": ret = size
-            case "M": ret = size * 1024
-            case "G": ret = size * 1024 * 1024
+            case "K" | "k": ret = size
+            case "M" | "m": ret = size * 1024
+            case "G" | "g": ret = size * 1024 * 1024
             case _:
                 raise Exception("Unexpected unit \"{}\".".format(pat.group("unit")))
 
@@ -72,7 +72,7 @@ def parse_function_graph_ftrace(trace_fpath: str):
     print("Shape: {}; head: \n{}".format(fngraph_df.shape, fngraph_df.head()))
     return fngraph_df
 
-def main(args: Namespace):
+def run(args: Namespace):
     # Load dataset -- each should have good 1000 entries
     ftrace_dat_paths: List[str] = []
     ftrace_dfs: List[Tuple(int, pd.DataFrame)] = []
@@ -113,7 +113,8 @@ def main(args: Namespace):
     fig.savefig(args.output)
     plt.show()
 
-if __name__ == "__main__":
+def main():
     args = parser.parse_args()
     sns.set_theme(context="talk", style="ticks", rc={"axes.grid": True})
-    main(args)
+    run(args)
+
diff --git a/src/aarch64-linux-flush-dcache/visualizer/out-log.pdf b/src/aarch64-linux-flush-dcache/visualizer/out-log.pdf
index b684569..6e7d34c 100644
Binary files a/src/aarch64-linux-flush-dcache/visualizer/out-log.pdf and b/src/aarch64-linux-flush-dcache/visualizer/out-log.pdf differ
diff --git a/src/aarch64-linux-flush-dcache/visualizer/pyproject.toml b/src/aarch64-linux-flush-dcache/visualizer/pyproject.toml
index 2060067..60d0eb6 100644
--- a/src/aarch64-linux-flush-dcache/visualizer/pyproject.toml
+++ b/src/aarch64-linux-flush-dcache/visualizer/pyproject.toml
@@ -11,7 +11,10 @@ matplotlib = "^3.8.3"
 seaborn = "^0.13.2"
 trappy = "^6.0.1"
 
+[tool.poetry.scripts]
+ftrace2pyplot = "ftrace2pyplot.core:main"
 
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+