Just found another great book for the A-1/B-1 lesson.
The “Knowabout” series has a book called “Sorting” by Henry Pluckrose & Chris Fairclough that is perfect. It poses a question on sorting or sets for each photo, then subsequent photos suggest ideas on how the items could be sorted, etc.