Styling improvements in UI