Add icons based on css class instead of regex