small ui changes
small ui changes