utils: zip_flat extended to take two args