Add start menu and restart screen