diff --git a/diy1.py b/diy1.py
index edb3c3c..1e0c934 100644
--- a/diy1.py
+++ b/diy1.py
@@ -51,8 +51,23 @@ print('    游戏开始')
 while True:
     print('-'*30)
     choice=input('请选择释放英雄技能（1攻击/2治疗）:')
-
-
+    if choice=q:
+        print('游戏结束')
+        break
+    elif choies=='1':
+        houyi.combat(yase)
+    elif choice=='2':
+        houyi.cure()
+    else:
+        print('请重新输入1或者2')
+        continue
+    status=random.randint(1,3)
+    if status==1:
+        yase.cure()
+    else:
+        yase.combat(houyi)            
+        
+