Update to restart command, adding to handler