diff --git a/test.py b/test.py
index d04439e..4355191 100644
--- a/test.py
+++ b/test.py
@@ -1,42 +1,31 @@
 import gpiozero
 import time
 from datetime import datetime
-import RPi.GPIO as GPIO
 
+def debug_print(message: str):
+    now = datetime.now()
+    current_time = now.strftime("%H:%M:%S")
+    print('[DEBUG]['+current_time+'] '+message)
 
-GPIO.setmode(GPIO.BOARD)
+def error_print(message: str):
+    now = datetime.now()
+    current_time = now.strftime("%H:%M:%S")
+    print('[ERROR]['+current_time+'] '+message)
 
-GPIO.setup(7, GPIO.OUT)
-GPIO.output(7, GPIO.HIGH)
-time.sleep(1)
-GPIO.output(7, GPIO.LOW)
-time.sleep(5)
-GPIO.cleanup()
+def relay_controller(pin: int, state: str):
+    if state == "True":
+        relay = gpiozero.OutputDevice(pin, active_high=False, initial_value=False)
+        relay.on()
+        debug_print("Set relay-pin "+str(pin)+" to state True")
+    elif state == "False":
+        relay = gpiozero.OutputDevice(pin, active_high=False, initial_value=False)
+        relay.off()
+        debug_print("Set relay-pin "+str(pin)+" to state False")
+    else:
+        error_print("Unknown state for relay ("+state+")")
 
-# def debug_print(message: str):
-#     now = datetime.now()
-#     current_time = now.strftime("%H:%M:%S")
-#     print('[DEBUG]['+current_time+'] '+message)
-
-# def error_print(message: str):
-#     now = datetime.now()
-#     current_time = now.strftime("%H:%M:%S")
-#     print('[ERROR]['+current_time+'] '+message)
-
-# def relay_controller(pin: int, state: str):
-#     if state == "True":
-#         relay = gpiozero.OutputDevice(pin, active_high=False, initial_value=False)
-#         relay.on()
-#         debug_print("Set relay-pin "+str(pin)+" to state True")
-#     elif state == "False":
-#         relay = gpiozero.OutputDevice(pin, active_high=False, initial_value=False)
-#         relay.off()
-#         debug_print("Set relay-pin "+str(pin)+" to state False")
-#     else:
-#         error_print("Unknown state for relay ("+state+")")
-
-# while True:
-#     relay_controller(7, "True")
-#     time.sleep(2)
-#     relay_controller(7, "False")
-#     time.sleep(2)
\ No newline at end of file
+while True:
+    relay_controller(4, "True")
+    time.sleep(2)
+    relay_controller(4, "False")
+    time.sleep(2)
\ No newline at end of file