add stage 2 solution