diff --git a/init.py b/init.py
index b605e42..eb2cda4 100644
--- a/init.py
+++ b/init.py
@@ -38,6 +38,9 @@ class AlarmClock:
         while True:
             self.loop()
 
+    def setRelayState(self, state: bool):
+        self.relayGpio.write(1 if state else 0)
+
     def loop(self):
         currentButtonState = not self.buttonGpio.read()
         if currentButtonState != self.lastKnownButtonState:
@@ -49,7 +52,7 @@ class AlarmClock:
         print(f"Button '{button}' pressed state: {state}")
         if button == "button":
             self.lastButtonPressTime = time()
-            self.relayGpio.write(1 if state else 0)
+            self.setRelayState(not self.relayGpio.read())
             self.setLcdText(self.lastButtonPressTime.__str__())
         
     def setLcdText(self, text: str):