feat : improve ui