Minor changes to stage 4.