Checkout - Improve UI part 2 updated