fix ui/ux bugs