slight updates to UI