diff --git a/scripts/cleanup b/scripts/cleanup
index e333437..0773d0f 100755
--- a/scripts/cleanup
+++ b/scripts/cleanup
@@ -2,5 +2,8 @@
 
 set -eu
 
+DEBIAN_FRONTEND=noninteractive apt-get -yq purge $(dpkg-query -W -f='${binary:Package}\n' 'linux-image-4*' 'linux-headers-4*' | grep -v $(uname -r))
+
 DEBIAN_FRONTEND=noninteractive apt-get -yq clean
+
 chsh --shell '/usr/sbin/nologin' packer
diff --git a/scripts/make_template b/scripts/make_template
index a4a9522..3d0e2c7 100755
--- a/scripts/make_template
+++ b/scripts/make_template
@@ -66,6 +66,11 @@ cat <<EOF | jq .
       ,"command": ". /tmp/venv/bin/activate && ANSIBLE_FORCE_COLOR=1 PYTHONUNBUFFERED=1 ansible-playbook"
       ,"playbook_file": "./ansible/{{ user \`box_name\` }}.yml"
     }
+    ,{
+      "type": "shell"
+      ,"execute_command": "sudo env {{ .Vars }} {{ .Path }}"
+      ,"inline": ["reboot"]
+    }
     ,{
       "type": "shell"
       ,"execute_command": "sudo env {{ .Vars }} {{ .Path }}"