minor change to UI