update for end screen