Wrote tests for stage model.