(feat) Styling code