diff --git a/core/html.py b/core/html.py
index d461af7..ed88585 100644
--- a/core/html.py
+++ b/core/html.py
@@ -1,4 +1,3 @@
-import  sys
 import  os
 
 class main:
@@ -8,10 +7,10 @@ class main:
 
     def get_html(location):
         html_dict = {}
-        html_page_list  = os.listdir( location )
+        html_page_list = os.listdir(location)
         for html_page in html_page_list:
-            full_path   = location + "/" + html_page
-            html_handle = open( full_path , 'r' )
-            html_raw    = html_handle.read()
-            html_dict[html_page] = html_raw
+            full_path = os.path.join(location, html_page)
+            if os.path.isfile(full_path):  # Ensure it's a file, not a directory
+                with open(full_path, 'r') as html_handle:
+                    html_dict[html_page] = html_handle.read()
         return html_dict