diff --git a/src/main.cpp b/src/main.cpp
index 1c4a514..d8030ce 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -15,23 +15,11 @@ void setup() {
 
 // Loop Section
 void loop() {
-  if (digitalRead(BUTTON) == LOW) {
-    digitalWrite(LED1, HIGH);
-    delay(250);
-    digitalWrite(LED1, LOW);
-
-    digitalWrite(LED2, HIGH);
-    delay(250);
-    digitalWrite(LED2, LOW);
-
-    digitalWrite(LED3, HIGH);
-    delay(250);
-    digitalWrite(LED3, LOW);
-
-  }
-  else {
-    digitalWrite(LED1, LOW);
-    digitalWrite(LED2, LOW);
-    digitalWrite(LED3, LOW);
+  while (digitalRead(BUTTON) == LOW) {
+    for (unsigned short int i = 3; i < 6; i++) {
+      digitalWrite(i, HIGH);
+      delay(250);
+      digitalWrite(i, LOW);
+    }
   }
 }