added TwoStageModel