Small refactor to remove these CSS classes in favor of tailwind. --------- Co-authored-by: wxiaoguang <[email protected]>