Add afterPayment()