trying to fix the display attribute method