Adding the load model to see if this captures what is needed.