add support for specifying classes for buttons