fix ui and design improvements