Experimenting with the numpy iterator