corrected the restart button and start screen