Updated the payment method logic