refactroed main display to div