1 files changed, 8 insertions, 2 deletions
diff --git a/src/mainwindow.h b/src/mainwindow.h
index 738d423..7d20a6e 100644
--- a/src/mainwindow.h
+++ b/src/mainwindow.h
@@ -11,6 +11,8 @@
 #include <QWidget>
 
 #include <QLightDM/Greeter>
+#include <QImage>
+#include <QRect>
 
 #include "loginform.h"
 
@@ -23,19 +25,23 @@ class MainWindow : public QWidget
     Q_OBJECT
 
 public:
-    explicit MainWindow(int screen, QWidget *parent = 0);
+    explicit MainWindow(bool primary, int screen, const QRect &rect, QWidget *parent = 0);
     ~MainWindow();
 
     void setFocus(Qt::FocusReason reason);
 
     bool showLoginForm();
 
-    LoginForm* loginForm() { return m_LoginForm;}
+    QImage& getBackground() { return m_background; }
+
+    LoginForm* loginForm() { return m_LoginForm; }
 private:
     int getOffset(QString offset, int maxVal, int defaultVal);
     void setBackground();
     int m_Screen;
+    bool m_Primary;
     LoginForm* m_LoginForm;
+    QImage m_background;
 };
 
 #endif // MAINWINDOW_H