Hooking up restart button