diff --git a/hyperionclient.h b/hyperionclient.h
index c58dd24..646de7e 100644
--- a/hyperionclient.h
+++ b/hyperionclient.h
@@ -2,6 +2,8 @@
 #define HYPERIONCLIENT_H
 
 #include <QTcpSocket>
+#include <QFile>
+#include <QIODevice>
 
 class HyperionClient : public QObject
 {
@@ -45,7 +47,8 @@ private:
     void _temperatureAdjustment();
 
 public slots:
-    void sendImage(QString);
+    void sendImage(const uchar *data, int size);
+    void setImgSize(int width, int height);
 };
 
 #endif // HYPERIONCLIENT_H