Updates to checkout process