MAINT: more work on toy model