improve the reboot script