Add support to apply class to any element (currently, only span is
supported)