diff --git a/src/main.cpp b/src/main.cpp
index f495601..030a606 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -11,45 +11,40 @@
 #include "src/Styles.hpp"
 #include <iostream>
 #include <memory>
+#include <sstream>
 #include <string>
 
+void update_count_label_text(std::shared_ptr<Raven::Label> label, int number) {
+    std::stringstream text;
+    text << "you have clicked the button " << std::to_string(number) << " times";
+    label->set_text(text.str());
+}
+
 int main() {
     Raven::Window window {};
+    int number = 0;
+
     window.spawn_window();
 
     auto main_widget = window.set_main_widget<Raven::Widget>();
     main_widget->set_layout<Raven::DocumentLayout>(10.0);
 
-    auto second_widget = main_widget->add<Raven::Widget>();
-    second_widget->set_layout<Raven::DocumentLayout>(20.0);
-    second_widget->resize(800, 800);
+    auto count_label = main_widget->add<Raven::Label>("you have clicked the button 0 times");
 
-    auto inner_widget = second_widget->add<Raven::Widget>();
-    inner_widget->set_layout<Raven::DocumentLayout>(8.0);
-    inner_widget->resize(600, 600);
+    main_widget->add<Raven::Widget>(Raven::ControlWidgetType::NewRow);
 
-    int number = 0;
+    auto increment_button = main_widget->add<Raven::Button>("click me");
+    increment_button->set_style(&Raven::accent_button_style);
+    increment_button->on_click = [&]() {
+        number++;
+        update_count_label_text(count_label, number);
+    };
 
-    for (int i = 0; i < 250; i++) {
-        auto button = inner_widget->add<Raven::Button>("click me");
-        button->set_style(&Raven::accent_button_style);
-        auto label = inner_widget->add<Raven::Label>("click one of the buttons!");
-        button->on_click = [&]() {
-            number += 10;
-
-            window.start_batch();
-            auto& children = inner_widget->children();
-            for (auto& child : children) {
-                if (child->type() == Raven::WidgetType::Button) {
-                    std::static_pointer_cast<Raven::Button>(child)->set_text(std::to_string(number));
-                }
-                if (child->type() == Raven::WidgetType::Label) {
-                    std::static_pointer_cast<Raven::Label>(child)->set_text(std::to_string(number));
-                }
-            }
-            window.end_batch();
-        };
-    }
+    auto reset_button = main_widget->add<Raven::Button>("reset");
+    reset_button->on_click = [&]() {
+        number = 0;
+        update_count_label_text(count_label, number);
+    };
 
     window.run(true);
     return 0;