small update on ui