few ui updates