Minor updates to UI