finished subsetting for first stage model