MAINT: work in progress -- more work with the switching process for the toy example