1 files changed, 18 insertions, 13 deletions
diff --git a/quick_lineup.py b/quick_lineup.py
index 237bc19..965922e 100644
--- a/quick_lineup.py
+++ b/quick_lineup.py
@@ -2,18 +2,23 @@ import sys
 from ql.console import Console
 
 
-if len(sys.argv) < 3 or len(sys.argv) > 4:
-    print('Give correct parameters: round, segment and (optionally) table')
-    sys.exit(1)
+def main():
+    if len(sys.argv) < 3 or len(sys.argv) > 4:
+        print('Give correct parameters: round, segment and (optionally) table')
+        sys.exit(1)
 
-round = int(sys.argv[1])
-segment = int(sys.argv[2])
-if len(sys.argv) == 4:
-    table = int(sys.argv[3])
-else:
-    table = None
+    round = int(sys.argv[1])
+    segment = int(sys.argv[2])
+    if len(sys.argv) == 4:
+        table = int(sys.argv[3])
+    else:
+        table = None
 
-try:
-    Console(round, segment, table).run()
-except:
-    print('ERROR: %s' % sys.exc_info()[1])
+    try:
+        Console(round, segment, table).run()
+    except:
+        print('ERROR: %s' % sys.exc_info()[1])
+
+
+if __name__ == '__main__':
+    main()