updating code so icons are displayed correctly