Updated final screen, added restart button functionality