Update to button styling