Defining the model's architecture