Added in some code for icons.