minor update on UI