fix ui/ux