minor enhances to UI