MAINT: more work in progress on this toy example