change to button to style nicely