diff --git a/src/main.cpp b/src/main.cpp
index 1dca956..1db32ae 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -98,19 +98,18 @@ void isArmed()
   size_t n = serializeJson(doc, buffer);
   client.publish("display/disarm", buffer, n);
   lcd.print("Armed");
-  delay(10000);
+  delay(5000);
   lcd.clear();
 }
 
 void loop()
 {
-  // digitalWrite(buzzerPin, HIGH);
-  // lcd.print(deviceId);
-  // delay(1000);
-  // lcd.clear();
-  // digitalWrite(buzzerPin, LOW);
-  // delay(1000);
-
-  client.loop(); // maybe?
+  client.loop();
+
+  if (!client.connected())
+  {
+    reconnect();
+  }
+  
   isArmed();
 }
\ No newline at end of file