Added a restart screen