Adding logic to create a display