updated display code