add function to set icon