diff --git a/scan.py b/scan.py
index 625c252..279e790 100644
--- a/scan.py
+++ b/scan.py
@@ -17,7 +17,7 @@ def get_ip():
         s.close()
     return IP
 
-if __name__ == "__main__":
+def main():
     ip = get_ip()
     print("IP:", ip)
     try:
@@ -44,3 +44,6 @@ if __name__ == "__main__":
     print("DONE")
     while True:
         time.sleep(10)
+
+if __name__ == "__main__":
+    main()