Switched from join to zip

- Tests use same partitioning
Improved documentation