working version with updated UI for checkout