diff --git a/demo/chatdialog.h b/demo/chatdialog.h
index 4c24335..b4c4b24 100644
--- a/demo/chatdialog.h
+++ b/demo/chatdialog.h
@@ -1,5 +1,8 @@
 #ifndef CHATDIALOG_H
 #define CHATDIALOG_H
+#include <boost/function.hpp>
+#include <vector>
+#include "digesttreescene.h"
 #include "ui_chatdialog.h"
 #include "chatbuf.pb.h"
 
@@ -9,17 +12,19 @@
 
 public:
 	ChatDialog(QWidget *parent = 0);
-
-public slots:
   void appendMessage(const SyncDemo::ChatMessage &msg);
-  void updateTreeView();
+  void processData(const std::vector<MisingDataInfo> &, SyncAppSocket *);
+
+private:
+  void formChatMessage(const QString &text, SyncDemo::ChatMessage &msg);
 
 private slots:
   void returnPressed();
-  void formChatMessage(const QString &text, SyncDemo::ChatMessage &msg);
 
 private:
   QString m_nick;
   QString m_chatroom;
+  QString m_prefix;
+  Sync::SyncAppSocket *m_sock;
 };
 #endif