Add reboot wrapper