diff --git a/src/Python/flask/web/app.py b/src/Python/flask/web/app.py
index 267a6e9..84b5dc1 100644
--- a/src/Python/flask/web/app.py
+++ b/src/Python/flask/web/app.py
@@ -5,18 +5,20 @@ import mysql.connector
 app = Flask(__name__)
 
 # Globale MQTT setup
-kobuki_message = "empty"
-def on_message(client, userdata, message):
-    global kobuki_message
-    kobuki_message = str(message.payload.decode("utf-8"))
-    print(kobuki_message)
+def on_message(client, message):
+    global kobuki_message, latest_image
+    if message.topic == "kobuki/data":
+        kobuki_message = str(message.payload.decode("utf-8"))
+    elif message.topic == "kobuki/cam":
+        latest_image = message.payload
 
+
+# Create an MQTT client instance
 mqtt_client = mqtt.Client()
 mqtt_client.username_pw_set("server", "serverwachtwoordofzo")
-mqtt_client.connect("localhost", 80, 60)
+mqtt_client.connect("localhost", 1884, 60)
 mqtt_client.loop_start()
 mqtt_client.subscribe("kobuki/data")
-mqtt_client.on_message = on_message
 
 # Database connectie-functie
 def get_db():