diff --git a/lab01/README.md b/lab01/README.md
index 3929de5e39cac6aefab48feba1e0de4e9242832f..f010f1cfefb332fd30a9f30ca7ecb33f2b258e97 100644
--- a/lab01/README.md
+++ b/lab01/README.md
@@ -135,13 +135,33 @@ client.loop_forever()
 ```python
 import paho.mqtt.client as mqtt
 import time
+
+flag_connected = 0
+def on_connect(client, userdata, flags, rc):
+   global flag_connected
+   flag_connected = 1
+
+def on_disconnect(client, userdata, rc):
+   global flag_connected
+   flag_connected = 0
+
 client = mqtt.Client()
+client.on_connect = on_connect
+client.on_disconnect = on_disconnect
 client.connect("localhost", 1883, 60)
+
+
+
 while(True):
-    client.publish("led/504", "1")
-    time.sleep(0.5)
-    client.publish("led/504", "0")
-    time.sleep(0.5)
+    if flag_connected == 1:
+        # Publish message
+        client.publish("led/504", "1")
+        time.sleep(1)
+        client.publish("led/504", "0")
+        time.sleep(1)
+    else:
+        # Wait to reconnect
+        client.reconnect()
 ```
 - Zadanie: Należy napisać oprogramowanie wg poniższego schamatu, które za pomocą serwera MQTT bedzie odbierac wiadomości z aplikacji button.py oraz odpowiednio nadwać wiadmosci do aplikacji led.py.
 ![Schamat zadania z MQTT](lab01/img/ztb_lab01_sch.png "Schamat zadania z MQTT")