updating the class to make the icons work