more changes to UI