simplify stage2.R