Display UI elements based on game state