updated stage 1 models with walls