Wrap buttons in divs to fix styles