Add reboot-required handling