1 files changed, 5 insertions, 2 deletions
diff --git a/include/ChatMessage.hpp b/include/ChatMessage.hpp
index 271aa5f..c13db9d 100644
--- a/include/ChatMessage.hpp
+++ b/include/ChatMessage.hpp
@@ -1,6 +1,6 @@
 #pragma once
 
-#include "types.hpp"
+#include <dchat/types.hpp>
 #include <gtkmm/grid.h>
 #include <gtkmm/label.h>
 
@@ -11,10 +11,13 @@ namespace dchat
     {
     public:
         ChatMessage(const Glib::ustring &username, const Glib::ustring &text, uint32_t timestampSeconds, const User *user);
-        
+
+        Gtk::Grid avatar;
         Gtk::Label username;
         Gtk::Label text;
         uint32_t timestampSeconds;
         const User *user;
+    private:
+        bool updateContent(const Cairo::RefPtr<Cairo::Context> &cairo);
     };
 }
 \ No newline at end of file