Update display code