Minor modification to UI