diff --git a/bench-boot b/bench-boot
index 68b2354..02888ac 100755
--- a/bench-boot
+++ b/bench-boot
@@ -62,6 +62,7 @@ fi
 
 arch=arm
 bench "$arch --eval '/poweroff.out'"
+newline
 if [ "$test_size" -ge 2 ]; then
   bench "$arch --eval '/poweroff.out' --trace exec_tb"
   qemu_insts "$arch"
diff --git a/common.py b/common.py
index bc1a0a4..7b4f394 100644
--- a/common.py
+++ b/common.py
@@ -777,6 +777,11 @@ Valid emulators: {}
         return imp.load_source(basename.replace('-', '_'), os.path.join(self.env['root_dir'], basename))
 
     def import_path_main(self, path):
+        '''
+        Import an object of the Main class of a given file.
+
+        By convention, we call the main object of all our CLI scripts as Main.
+        '''
         return self.import_path(path).Main()
 
     def log_error(self, msg):
@@ -795,7 +800,7 @@ Valid emulators: {}
         env = kwargs.copy()
         self.input_args = env.copy()
         env.update(consts)
-        real_all_archs= env['all_archs']
+        real_all_archs = env['all_archs']
         if real_all_archs:
             real_archs = consts['all_long_archs']
         else:
diff --git a/trace-boot b/trace-boot
index 6b6a8f7..ae29648 100755
--- a/trace-boot
+++ b/trace-boot
@@ -1,63 +1,62 @@
 #!/usr/bin/env python3
 
-import imp
-import os
-import subprocess
-import re
-
 import common
-run = imp.load_source('run', os.path.join(kwargs['root_dir'], 'run'))
-qemu_trace2txt = imp.load_source('qemu_trace2txt', os.path.join(kwargs['root_dir'], 'qemu-trace2txt'))
+from shell_helpers import LF
 
-parser = self.get_argparse(argparse_args={
-    'description': '''Trace the PIC addresses executed on a Linux kernel boot.
+class Main(common.LkmcCliFunction):
+    def __init__(self):
+        super().__init__(
+            description='''Trace the PIC addresses executed on a Linux kernel boot.
 
 More information at: https://github.com/cirosantilli/linux-kernel-module-cheat#tracing
 '''
-})
-parser.add_argument(
-    'extra_emulator_args', nargs='*',
-    help='Extra options to append at the end of the emulator command line'
-)
-args = self.setup(parser)
-extra_args = {
-    'extra_emulator_args': kwargs['extra_emulator_args'],
-}
-if kwargs['emulator'] == 'gem5':
-    extra_args.update({
-        'eval': 'm5 exit',
-        'trace': 'Exec,-ExecSymbol,-ExecMicro',
-    })
-    run.main(args, extra_args)
-else:
-    extra_args.update({
-        'kernel_cli': 'init=/poweroff.out',
-        'trace': 'exec_tb',
-    })
-    run.main(args, extra_args)
-    qemu_trace2txt.main()
-    # Instruction count.
-    # We could put this on a separate script, but it just adds more arch boilerplate to a new script.
-    # So let's just leave it here for now since it did not add a significant processing time.
-    kernel_entry_addr = hex(self.get_elf_entry(kwargs['vmlinux']))
-    nlines = 0
-    nlines_firmware = 0
-    with open(kwargs['qemu_trace_txt_file'], 'r') as trace_file:
-        in_firmware = True
-        for line in trace_file:
-            line = line.rstrip()
-            nlines += 1
-            pc = line.split('=')[-1]
-            if pc == kernel_entry_addr:
-                in_firmware = False
-            if in_firmware:
-                nlines_firmware += 1
-    print('''\
+        )
+
+    def timed_main(self):
+        args = self.get_common_args()
+        run = self.import_path_main('run')
+        if self.env['emulator'] == 'gem5':
+            args.update({
+                'eval': 'm5 exit',
+                'trace': 'Exec,-ExecSymbol,-ExecMicro',
+            })
+            run.main(**args)
+        elif self.env['emulator'] == 'qemu':
+            run_args = args.copy()
+            run_args.update({
+                'kernel_cli': 'init=/poweroff.out',
+                'trace': 'exec_tb',
+            })
+            run.main(**run_args)
+            qemu_trace2txt = self.import_path_main('qemu-trace2txt')
+            qemu_trace2txt.main(**args)
+            # Instruction count.
+            # We could put this on a separate script, but it just adds more arch boilerplate to a new script.
+            # So let's just leave it here for now since it did not add a significant processing time.
+            kernel_entry_addr = hex(self.get_elf_entry(self.env['vmlinux']))
+            nlines = 0
+            nlines_firmware = 0
+            with open(self.env['qemu_trace_txt_file'], 'r') as trace_file:
+                in_firmware = True
+                for line in trace_file:
+                    line = line.rstrip()
+                    nlines += 1
+                    pc = line.split('=')[-1]
+                    if pc == kernel_entry_addr:
+                        in_firmware = False
+                    if in_firmware:
+                        nlines_firmware += 1
+            print('''\
 instructions {}
 entry_address {}
-instructions_firmware {}\
+instructions_firmware {}
 '''.format(
-        nlines,
-        kernel_entry_addr,
-        nlines_firmware
-    ))
+                    nlines,
+                    kernel_entry_addr,
+                    nlines_firmware
+                ),
+                end=''
+            )
+
+if __name__ == '__main__':
+    Main().cli()