Added updated display back into the game