Add re-run button