Added support for rebooting