Add Restart button