diff --git a/defaults/main.yml b/defaults/main.yml
index e8b6a5d..6973eb7 100644
--- a/defaults/main.yml
+++ b/defaults/main.yml
@@ -1,2 +1,3 @@
 ---
-update_autoremove: yes
\ No newline at end of file
+update_autoremove: yes
+update_reboot_allowed: yes
\ No newline at end of file
diff --git a/tasks/main.yml b/tasks/main.yml
index 655d41f..6ded82f 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -8,6 +8,25 @@
 - name: "Autoremove and cleanup"
   apt:
     autoremove: yes
-    clean: yes
+    autoclean: yes
   when:
-    - "update_autoremove"
\ No newline at end of file
+    - "update_autoremove"
+
+- name: "Check if reboot file exists"
+  stat:
+    path: "/var/run/reboot-required"
+  register: p
+
+- name: "Output required reboot"
+  debug:
+    msg:
+    - "Reboot required: {{ p.stat.exists }}"
+    - "Reboot allowed: {{ update_reboot_allowed }}"
+
+- name: "Reboot Host"
+  reboot:
+    msg: "Reboot initiated by Ansible"
+    post_reboot_delay: 10
+  when:
+    - "p.stat.exists"
+    - "update_reboot_allowed"
\ No newline at end of file
diff --git a/tests/test.yml b/tests/test.yml
index 4857cf8..22cb392 100644
--- a/tests/test.yml
+++ b/tests/test.yml
@@ -1,5 +1,5 @@
 ---
-- hosts: localhost
+- hosts: all
   remote_user: root
   roles:
     - Ansible-Role-Update
\ No newline at end of file