add style and better ux