Add function to add icon states