diff --git a/square.py b/square.py
index f1fb2a2..8c3c450 100644
--- a/square.py
+++ b/square.py
@@ -1,17 +1,10 @@
 from turtle import *
 
 def square(side_length):
-    forward(side_length)
-    right(90)
-    forward(side_length)
-    right(90)
-    forward(side_length)
-    right(90)
-    forward(side_length)
-    right(90)
-
-sizes = [20, 40, 60, 80, 100]
-for size in sizes:
-    square(size)
-input()
-
+    for i in [1,2,3,4]:
+        forward(side_length)
+        right(90)
+square(200)
+square(150)
+square(100)
+square(50)