Modify restart button