Surprise_CEI_with_step