add stage based on S1E2