Modify the load model example