Small updates to the UI