minor changes to UI