added ability to change the class of the buttons

changed the model to be able to also set the class of the buttons