diff --git a/lab-01/r-1.4_scratch.py b/lab-01/r-1.4_scratch.py
index 6a35776..fa379d0 100644
--- a/lab-01/r-1.4_scratch.py
+++ b/lab-01/r-1.4_scratch.py
@@ -15,10 +15,9 @@ def squares_sum(n):
         n = n - 1
         list.append(n)
         if n == 1:
-            square_list = []
-            for val in list:
-                square_list.append(val*val)
-                print(square_list)
+            square_list = [m*m for m in list]
+            print(square_list)
+