fix button display
work on styling