UI styling fixes