1 files changed, 8 insertions, 2 deletions

diff --git a/aobutton.h b/aobutton.h
index 72bebf8..f0c5697 100644
--- a/aobutton.h
+++ b/aobutton.h
@@ -1,11 +1,17 @@
 #ifndef AOBUTTON_H
 #define AOBUTTON_H
 
+#include <QPushButton>
 
 class AOButton : public QPushButton
 {
+  Q_OBJECT
+
 public:
-  AOButton();
+  AOButton(QWidget *parent);
+  ~AOButton();
+
+  void set_image(QString p_image);
 };
 
-#endif // AOBUTTON_H
\ No newline at end of file
+#endif // AOBUTTON_H