more work on the toy model example