Modified the display function