diff --git a/demo/chatdialog.h b/demo/chatdialog.h
index 401df9d..0f65426 100644
--- a/demo/chatdialog.h
+++ b/demo/chatdialog.h
@@ -1,6 +1,7 @@
 #ifndef CHATDIALOG_H
 #define CHATDIALOG_H
 #include "ui_chatdialog.h"
+#include "chatbuf.pb.h"
 
 class ChatDialog : public QDialog,  private Ui::ChatDialog 
 {
@@ -8,5 +9,17 @@
 
 public:
 	ChatDialog(QWidget *parent = 0);
+
+public slots:
+  void appendMessage(SyncDemo::ChatMessage &msg);
+  void updateTreeView();
+
+private slots:
+  void returnPressed();
+  void formChatMessage(const QString &text, SyncDemo::ChatMessage &msg);
+
+private:
+  QString m_nick;
+  QString m_chatroom;
 };
 #endif