diff --git a/main.py b/main.py
index f9d1d26..cae5e21 100644
--- a/main.py
+++ b/main.py
@@ -32,7 +32,11 @@ def download():
 
         # print(base[y]+url[y])
         query = """ select active from manga where manga_id = """ + str(y+1)
-        active = con.execute(query).fetchone()
+        try:
+            active = con.execute(query).fetchone()
+        except engine.Error as e:
+            if 'MySQL server has gone away' in str(e):
+                con = engine.connect()
 
         if active[0] != 0:
             print("""manga-py """ + base[y]+url[y] + """ -n \"""" + name[y] + """\" -d \"""" + PATH + """\" -z -g """)
@@ -80,7 +84,7 @@ def upload_jason():
             z=z+1
 
 while True:
-    print("1. Download \n 2. Upload")
+    print("1. Download \n 2. Upload\n 3. Exit")
     val = input(">>>")
     if val == "1":
         download()