Attempting to move the display elements to use an OO approach