Added after_success argument