Fixing legend and text, adding restart link