diff --git a/src/test/data/pa1/student_contributed/good.py b/src/test/data/pa1/student_contributed/good.py
index 4c01c88..88f20ca 100644
--- a/src/test/data/pa1/student_contributed/good.py
+++ b/src/test/data/pa1/student_contributed/good.py
@@ -74,4 +74,15 @@ Here I am"
 """
 This is a
 multi-line comment.
-"""
\ No newline at end of file
+"""
+
+
+def get_stones(name:str)->str:
+    color=['Red','Blue']
+    stones=['Mind','Soul']
+    def map_name(nm:str)->str:
+        return stones[color.index(nm)]
+    return map_name(name)
+
+stone="Blue"
+print(get_stones(stone))
\ No newline at end of file