Refactor internal code towards a better atomic design system

[Anton Kalmanovich]

Reference idea: https://github.com/radix-ui/design-system/tree/master/components