fix UI styling