diff --git a/国庆考核.py b/国庆考核.py
index 47b4e17..328be32 100644
--- a/国庆考核.py
+++ b/国庆考核.py
@@ -12,4 +12,8 @@ class Square(Shape):
     def __init__(self, side):
         self.side = side
     def area(self):
-        return self.side * self.side
\ No newline at end of file
+        return self.side * self.side
+
+r = int(input("请输入圆的半径: "))
+side = int(input("请输入正方形的边长: "))
+shapes = [Circle(r), Square(side)]