I thought about that, but you need to be able to call the expand function in a final flat map, which peeks inside alas. couldn't seem to get around that need in a great way