Added icon functionality