small ui changes

based on review