diff --git a/diy.py b/diy.py
index e586ff4..6eae575 100644
--- a/diy.py
+++ b/diy.py
@@ -4,12 +4,12 @@ class Hero:
         self.level=1
         self.hp=hp
         self.attack=attack
-def upgrade():
-    yase.level=yase.level+1
-    yase.hp=yase.hp+40
-    yase.attack=yase.attack+9000
+    def upgrade(self):
+        self.level=self.level+1
+        self.hp=self.hp+40
+        self.attack=self.attack+9000
 yase=Hero("亚瑟",120,40)
-print("亚瑟的血量是：",yase.hp)
-upgrade()
-upgrade()
-print("升级两次后亚瑟的血量是：",yase.hp)
+yase.upgrade()
+print("升级1次后亚瑟的等级是：",yase.level)
+print("升级1次后亚瑟的血量是：",yase.hp)
+print("升级1次后亚瑟的攻击力是：",yase.attack)
\ No newline at end of file