Made restart better for graphics

Same as the title