Add after_success thing