2 files changed, 5 insertions, 3 deletions
diff --git a/termcast_server/__init__.py b/termcast_server/__init__.py
index 3a8dfee..6896b04 100644
--- a/termcast_server/__init__.py
+++ b/termcast_server/__init__.py
@@ -76,3 +76,7 @@ class Server(object):
         sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
         sock.bind(('', port))
         return sock
+
+def main():
+    server = Server(sys.argv[1])
+    server.listen()
diff --git a/termcast_server/__main__.py b/termcast_server/__main__.py
index d842d0a..b4a831a 100644
--- a/termcast_server/__main__.py
+++ b/termcast_server/__main__.py
@@ -1,5 +1,3 @@
-import sys
 import termcast_server
 
-server = termcast_server.Server(sys.argv[1])
-server.listen()
+termcast_server.main()