more corrections for stage 3
more corrections for stage 3