From aa98d610b0afb2b9ebf174f54b090c7722c65041 Mon Sep 17 00:00:00 2001
From: Andreas Zweili <andreas@zweili.ch>
Date: Sun, 19 Dec 2021 18:48:18 +0100
Subject: [PATCH] extend the main function

---
 scripts/format-disk.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/format-disk.py b/scripts/format-disk.py
index 6f579ae..76816d3 100644
--- a/scripts/format-disk.py
+++ b/scripts/format-disk.py
@@ -42,6 +42,9 @@ def encrypt_disk():
 
 def main():
     disks = read_disks()
+    create_menu(disks)
+    disk_to_format = disks[get_disk_to_format()]
+    format_disk(disk_to_format)
 
 
 if __name__ == "__main__":