a few changes to UI