Styling fix, added CSS class to button