Adding models for the stages of the cascade