diff --git a/day4/TODO.py b/day4/TODO.py
index 5936270..15cf818 100644
--- a/day4/TODO.py
+++ b/day4/TODO.py
@@ -1,16 +1,15 @@
-list=['原神']
-for i in list:
-    list.append(input('何事？'))
-    a=input('何事终了？')
+list = ["原神"]
+while True:
+    list.append(input("何事？"))
+    a = input("何事终了？")
     if a in list:
         list.remove(a)
     else:
         pass
-    print('未尽之事：',list)
-    b = input('是否下一周目(1是2否)')
-    if b == 1:
-        continue;
+    print("未尽之事：", list)
+    b = input("是否下一周目(1是2否)")
+    if b == "1":
+        continue
     else:
-        print('未尽之事：',list)
-        break;
-#为什么没用到i也可以用i来进行循环
\ No newline at end of file
+        print("未尽之事：", list)
+        break
\ No newline at end of file