Add after success block