Made changes to the checkout decorator