1 files changed, 4 insertions, 2 deletions
diff --git a/include/aoemotebutton.h b/include/aoemotebutton.h
index add9b43..edabfb6 100644
--- a/include/aoemotebutton.h
+++ b/include/aoemotebutton.h
@@ -5,6 +5,7 @@
 #include <QPainter>
 #include <QDebug>
 #include <QPushButton>
+#include <QLabel>
 
 class AOEmoteButton : public QPushButton {
   Q_OBJECT
@@ -16,16 +17,17 @@ public:
   void set_image(QString p_image, QString p_emote_comment);
   void set_char_image(QString p_char, int p_emote, bool on);
 
+  void set_selected_image(QString p_image);
+
   void set_id(int p_id) { m_id = p_id; }
   int get_id() { return m_id; }
 
 private:
   QWidget *parent;
   AOApplication *ao_app;
+  QLabel *ui_selected = nullptr;
 
   int m_id = 0;
-
-
 signals:
   void emote_clicked(int p_id);