diff --git a/doc_resources/gateway/class_diagram.puml b/doc_resources/gateway/class_diagram.puml
index 484dd9e..78c6ad2 100644
--- a/doc_resources/gateway/class_diagram.puml
+++ b/doc_resources/gateway/class_diagram.puml
@@ -5,17 +5,20 @@ class Gateway {
     - gateway_id: str
     - mqtt_broker: str
     - mqtt_port: int
-    - service_uuid: str
     - mqttc: MQTTClient
     --
     + KEY_WINDOW: 0x01
     + KEY_HUMIDITY: 0x02
     + KEY_TEMP: 0x03
     + KEY_CO2: 0x04
+    + INVALID_VALUE: 0xFFFFFFFF
+    + EXPECTED_PAYLOAD_SIZE: 12
     --
     + decode_payload(data: bytes): dict
     + publish(mac: str, data: dict)
     + on_device_found(device, adv_data)
+    + on_connect(client, userdata, flags, reason_code, properties)
+    + on_publish(client, userdata, mid, reason_code, properties)
     + run()
 }
 
@@ -29,6 +32,7 @@ class MQTTClient {
     + publish()
     + tls_set()
     + username_pw_set()
+    + loop_start()
 }
 
 Gateway o-- BleakScanner