#3 adding class on RenderButtonTrait.php