adding end of the game and restart