Updated model with Hospital included in the loop