diff --git a/src/test/data/pa1/student_contributed/good.py b/src/test/data/pa1/student_contributed/good.py
index 88f20ca..093076f 100644
--- a/src/test/data/pa1/student_contributed/good.py
+++ b/src/test/data/pa1/student_contributed/good.py
@@ -1,21 +1,34 @@
+class Foo(object):
+    x:int = 0
 
-print("Hi")
-def fun1():
-    def fun2():
-        print("Hello")
-    fun2()
+    def __init__(self:"Foo", x:int):
+        self.x = x
     
-print("World")
+    def bar(y:int):
+        print("Hello World!",self.x+y)
+        y = 10
 
-def fun3():
-    def fun4():
+def get_stones(name:str)->str:
+    def map_name(nm:str)->str:
+        return stones[color.index(nm)]
+    color=["Red","Blue"]
+    stones=["Mind","Soul"]
+    return map_name(name)
+
+def funa():
+    def funb():
+        print("Hello")
+    funb()
+
+def fund():
+    def fune():
         print("Hello")
     c = 4 + 5
 
-def fun5():
-    c = 6
-    def fun6():
+def funf():
+    def fung():
         print("Hello")
+    c = 6
     c = 4 + 5
 
 
@@ -51,38 +64,17 @@ if True:
 else:
     print("World")
 
-class Foo(object):
-    x:int = 0
 
-    def __init__(self:"Foo", x:int):
-        self.x = x
-    
-    def bar(y:int)->None:
-        print("Hello World!",self.x+y)
-        y = 10
 
 f = Foo(1)
 print(f.x)
 f.bar(4)
 
-a=[[[1],[2]][[3],[4]]]
+a=[[[1],[2]],[[3],[4]]]
 print(a[0][0][1]*a[1][1][0])
 
 multiline_string="Hi World,
 Here I am"
 
-"""
-This is a
-multi-line comment.
-"""
-
-
-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