Add conditional success message