From cf1d11eb789280ddf9eb44aac2cb37363a8460be Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Sun, 14 Apr 2019 17:16:13 -0700
Subject: [PATCH] Always show header

---
 .bin/Scripts/build-ufd | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/.bin/Scripts/build-ufd b/.bin/Scripts/build-ufd
index 38b6f8d6..592a6143 100755
--- a/.bin/Scripts/build-ufd
+++ b/.bin/Scripts/build-ufd
@@ -42,6 +42,19 @@ def get_user_name():
 
 # Main section
 if __name__ == '__main__':
+  # Set log
+  try:
+    global_vars['LogDir'] = '{}/Logs'.format(
+      get_user_home(get_user_name()))
+    set_log_file('Build UFD ({Date-Time}).log'.format(**global_vars))
+  except:
+    major_exception()
+
+  # Header
+  print_success(KIT_NAME_FULL)
+  print_standard('UFD Build Tool')
+  print_standard(' ')
+
   # Check if running as root
   if os.geteuid() != 0:
     print_error('ERROR: This script is meant to be run as root.')
@@ -58,11 +71,6 @@ if __name__ == '__main__':
     major_exception()
 
   try:
-    # Set log
-    global_vars['LogDir'] = '{}/Logs'.format(
-      get_user_home(get_user_name()))
-    set_log_file('Build UFD ({Date-Time}).log'.format(**global_vars))
-
     # Verify selections
     sources = OrderedDict()
     ## UFD
@@ -90,11 +98,6 @@ if __name__ == '__main__':
         sources[label] = s_path_obj
 
     # Show selections
-    ## Header
-    clear_screen()
-    print_success(KIT_NAME_FULL)
-    print_standard('UFD Build Tool')
-    print_standard(' ')
     ## Sources
     print_info('Sources')
     for label in UFD_SOURCES.keys():
@@ -145,7 +148,7 @@ if __name__ == '__main__':
 
     print_standard(' ')
     print_success("It's go-time!")
-    exit_script(1)
+    exit_script()
 
     # Double-check if formating device