Add ability to add wrapper with custom class for element