Implemented better restart screens