1 files changed, 9 insertions, 2 deletions
diff --git a/src/DisplayApp/Screens/Message.cpp b/src/DisplayApp/Screens/Message.cpp
index c9e0938..c8a4ea1 100644
--- a/src/DisplayApp/Screens/Message.cpp
+++ b/src/DisplayApp/Screens/Message.cpp
@@ -19,7 +19,7 @@ static void event_handler(lv_obj_t * obj, lv_event_t event) {
   screen->OnObjectEvent(obj, event);
 }
 
-Message::Message(DisplayApp* app, Pinetime::Components::Gfx &gfx) : Screen(app, gfx) {
+Message::Message(DisplayApp* app) : Screen(app) {
 
   backgroundLabel = lv_label_create(lv_scr_act(), NULL);
   backgroundLabel->user_data = this;
@@ -55,11 +55,13 @@ Message::~Message() {
   lv_obj_clean(lv_scr_act());
 }
 
-void Message::Refresh(bool fullRefresh) {
+bool Message::Refresh() {
   if(previousClickCount != clickCount) {
     lv_label_set_text_fmt(labelClick, "%d", clickCount);
     previousClickCount = clickCount;
   }
+
+  return running;
 }
 
 void Message::OnObjectEvent(lv_obj_t *obj, lv_event_t event) {
@@ -79,3 +81,8 @@ void Message::OnObjectEvent(lv_obj_t *obj, lv_event_t event) {
     NRF_LOG_INFO("Toggled");
   }
 }
+
+bool Message::OnButtonPushed() {
+  running = false;
+  return true;
+}