Updated restart screen - fixed a few bugs should almost be solid