allow custom class on buttons
fix