Updated CSS class of button in edit