Designing the architecture for the model