diff --git a/src/main.cpp b/src/main.cpp
index 9595c9b..edd954e 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1,17 +1,25 @@
 #include <Arduino.h>
 
+#define LED 4
+#define BUTTON1 2
+#define BUTTON2 3
+
 // Setup Section
 void setup() {
-  pinMode(13, OUTPUT);
-  pinMode(12, OUTPUT);
-  pinMode(11, OUTPUT);
+  pinMode(BUTTON1, INPUT);
+  pinMode(BUTTON2, INPUT);
+  pinMode(LED, OUTPUT);
 }
 
 // Loop Section
 void loop() {
-  for (int i = 13; i > 10; i--) {
-    digitalWrite(i, HIGH);
-    delay(1000);
-    digitalWrite(i, LOW);
+  if (digitalRead(BUTTON1) == LOW || digitalRead(BUTTON2) == LOW) {
+    // check the lamp is ON or OFF
+    if (digitalRead(LED) == HIGH) {
+      // add print here please
+      digitalWrite(LED, LOW);
+    } else {
+      digitalWrite(LED, HIGH);
+    }
   }
 }