Add Restart() handler