add option to add classes to buttons