Some updates to UI styles