feat: Improve UI on checkout page