1 files changed, 15 insertions, 7 deletions
diff --git a/project2/proj2_s4498062/dns_server.py b/project2/proj2_s4498062/dns_server.py
index 07e0d9a..4ac2ec4 100644
--- a/project2/proj2_s4498062/dns_server.py
+++ b/project2/proj2_s4498062/dns_server.py
@@ -7,16 +7,21 @@ This script contains the code for starting a DNS server.
 
 import dns.server
 
-if __name__ == "__main__":
+
+def main():
+    """DNS server"""
     # Parse arguments
     import argparse
     parser = argparse.ArgumentParser(description="DNS Server")
-    parser.add_argument("-c", "--caching", action="store_true",
-            help="Enable caching")
-    parser.add_argument("-t", "--ttl", metavar="time", type=int, default=0, 
-            help="TTL value of cached entries (if > 0)")
-    parser.add_argument("-p", "--port", type=int, default=5353,
-            help="Port which server listens on")
+    parser.add_argument(
+        "-c", "--caching", action="store_true",
+        help="Enable caching")
+    parser.add_argument(
+        "-t", "--ttl", metavar="time", type=int, default=0,
+        help="TTL value of cached entries (if > 0)")
+    parser.add_argument(
+        "-p", "--port", type=int, default=5353,
+        help="Port which server listens on")
     args = parser.parse_args()
 
     # Start server
@@ -26,3 +31,6 @@ if __name__ == "__main__":
     except KeyboardInterrupt:
         server.shutdown()
         print()
+
+if __name__ == "__main__":
+    main()