diff --git a/square.py b/square.py
index f1fb2a2..646b099 100644
--- a/square.py
+++ b/square.py
@@ -1,14 +1,12 @@
 from turtle import *
 
+sides = [1,2,3,4]
 def square(side_length):
-    forward(side_length)
-    right(90)
-    forward(side_length)
-    right(90)
-    forward(side_length)
-    right(90)
-    forward(side_length)
-    right(90)
+    for side in sides:
+        forward(20)
+        right(90)
+    
+    
 
 sizes = [20, 40, 60, 80, 100]
 for size in sizes: