Added main function and conditional execution

1 files changed, 12 insertions, 8 deletions

diff --git a/archey.new b/archey.new
index e7dcd3a..ec72334 100644
--- a/archey.new
+++ b/archey.new
@@ -320,11 +320,15 @@ class User(BaseDisplay):
 
 
 ## TEST ## <<< TEMPORARY 
-out = Output()
-out.append(Shell())
-out.append(User())
-out.append(Terminal())
-out.append(Uptime())
-out.append(WindowManager())
-#out.append(DesktopEnvirornment())
-out.output()
+def main():
+	out = Output()
+	out.append(Shell())
+	out.append(User())
+	out.append(Terminal())
+	out.append(Uptime())
+	out.append(WindowManager())
+	#out.append(DesktopEnvirornment())
+	out.output()
+
+if __name__ == '__main__':
+	main()