Adding support for shutdown and reboot