Made partition assignment strategy configurable
Updated JavaDoc for partitioning.