diff --git a/main.py b/main.py
index 18cfd2b..90c3c40 100644
--- a/main.py
+++ b/main.py
@@ -2,19 +2,19 @@ import requests
 from bs4 import BeautifulSoup
 
 # CANVAS_URL = "https://canvas.auckland.ac.nz"
-CANVAS_URL = "https://canvas.auckland.ac.nz/courses/61920/modules"
+COURSE_ID = "61920"
+CANVAS_URL = "https://canvas.auckland.ac.nz"
 
 def read_cookie():
     with open("cookie.txt", "r") as file:
         cookie = file.read().strip()
     return cookie
     
-    
-if __name__ == "__main__":
+def main():
     cookie = read_cookie()
     
     r = requests.get(
-        url=CANVAS_URL,
+        url=f"{CANVAS_URL}/courses/{COURSE_ID}/modules",
         headers={
             "Cookie": cookie,
         })
@@ -23,4 +23,7 @@ if __name__ == "__main__":
     
     links = soup.findAll("a", {"class": "ig-title title item_link"})
     print(links)
+    
+if __name__ == "__main__":
+    main()
     
\ No newline at end of file