small updates to UI